Hugging Face's logo

Spaces:
keras-dreambooth
/
dreambooth_dosa
Runtime error

App Files Community
bharat-raghunathan commited on
Commit
0b1e829
·
1 Parent(s): 6df4440

Add model link

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ keras.mixed_precision.set_global_policy("mixed_float16")
8
  sd_dreambooth_model = models.StableDiffusion(
9
  img_width=512, img_height=512, jit_compile=True
10
  )
11
- db_diffusion_model = from_pretrained_keras("keras-dreambooth/dreambooth_dosa")
12
  sd_dreambooth_model._diffusion_model = db_diffusion_model
13
 
14
  # generate images
 
8
  sd_dreambooth_model = models.StableDiffusion(
9
  img_width=512, img_height=512, jit_compile=True
10
  )
11
+ db_diffusion_model = from_pretrained_keras("keras-dreambooth/dreambooth_dosa_v2")
12
  sd_dreambooth_model._diffusion_model = db_diffusion_model
13
 
14
  # generate images