Update README.md
Browse files
README.md
CHANGED
@@ -44,8 +44,8 @@ import numpy as np
|
|
44 |
|
45 |
model_id = "runwayml/stable-diffusion-v1-5"
|
46 |
|
47 |
-
unet = UNet2DConditionModel.from_pretrained("ckandemir/
|
48 |
-
text_encoder = CLIPTextModel.from_pretrained("ckandemir/
|
49 |
|
50 |
pipeline = StableDiffusionPipeline.from_pretrained(
|
51 |
model_id, unet=unet, text_encoder=text_encoder, dtype=torch.float16, use_safetensors=True
|
|
|
44 |
|
45 |
model_id = "runwayml/stable-diffusion-v1-5"
|
46 |
|
47 |
+
unet = UNet2DConditionModel.from_pretrained("ckandemir/boredape_diffusion", subfolder="unet")
|
48 |
+
text_encoder = CLIPTextModel.from_pretrained("ckandemir/boredape_diffusion",subfolder="text_encoder")
|
49 |
|
50 |
pipeline = StableDiffusionPipeline.from_pretrained(
|
51 |
model_id, unet=unet, text_encoder=text_encoder, dtype=torch.float16, use_safetensors=True
|