nielsgl commited on
Commit
bb8b10f
·
1 Parent(s): ba2fab8

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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])