vyroAI
/

ImagineV4.1

StableDiffusionPipeline

Model card Files Files and versions

rashidvyro commited on Dec 31, 2023

Commit

dee524f

·

1 Parent(s): af2f56c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -61,7 +61,7 @@ pip install diffusers transformers accelerate --upgrade
 - **Text-to-image**:
 ```py
-from diffusers import AutoPipelineForText2Image
 import torch
 pipe = AutoPipelineForText2Image.from_pretrained("vyroAI/ImagineV4.1", torch_dtype=torch.float16, variant="fp16")

 - **Text-to-image**:
 ```py
+from diffusers import stable_diffusion_custom_v4_1
 import torch
 pipe = AutoPipelineForText2Image.from_pretrained("vyroAI/ImagineV4.1", torch_dtype=torch.float16, variant="fp16")