Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
@@ -73,8 +73,8 @@ import matplotlib.pyplot as plt
|
|
73 |
|
74 |
|
75 |
model = keras_cv.models.StableDiffusion(img_width=512, img_height=512, jit_compile=True)
|
76 |
-
model._diffusion_model = from_pretrained_keras(nielsgl/dreambooth-pug-ace)
|
77 |
-
model._text_encoder = from_pretrained_keras(nielsgl/dreambooth-pug-ace-text-encoder)
|
78 |
|
79 |
images = model.text_to_image("a photo of puggieace dog on the beach", batch_size=3)
|
80 |
plt.imshow(image[0])
|
|
|
73 |
|
74 |
|
75 |
model = keras_cv.models.StableDiffusion(img_width=512, img_height=512, jit_compile=True)
|
76 |
+
model._diffusion_model = from_pretrained_keras("nielsgl/dreambooth-pug-ace")
|
77 |
+
model._text_encoder = from_pretrained_keras("nielsgl/dreambooth-pug-ace-text-encoder")
|
78 |
|
79 |
images = model.text_to_image("a photo of puggieace dog on the beach", batch_size=3)
|
80 |
plt.imshow(image[0])
|