Instructions to use haoningwu/StoryGen with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use haoningwu/StoryGen with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("haoningwu/StoryGen", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Unable to load the model using diffusers
#1
by AradhyaM - opened
Upon loading the model using diffusers I get the following error.
from diffusers import DiffusionPipeline
pipeline = DiffusionPipeline.from_pretrained("haoningwu/StoryGen")
EntryNotFoundError: 404 Client Error. (Request ID: Root=1-6628e36a-5029d4521e1ad4fd4c75815c;dafb8b6f-48fb-4971-ba01-8918da8dcc41)
Entry Not Found for url: https://huggingface.co/haoningwu/StoryGen/resolve/main/model_index.json.`
Well, our model has not been integrated into DiffusionPipeline of diffusers, we recommend you to use the code in our GitHub repository: https://github.com/haoningwu3639/StoryGen, to load our pre-trained models.