Hugging Face's logo

Spaces:
keras-dreambooth
/
keras-dreambooth-riffusion-currulao
Runtime error

App Files Community
2
juancopi81 commited on
Commit
a4f9fc2
•
1 Parent(s): f3d8154

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights
15
 
16
  sd_dreambooth_model_1.load_weights_from_pytorch_ckpt("riffusion-model-v1.ckpt")
17
 
18
- sd_dreambooth_model_1.diffusion_model.load_weights("dreambooth_riffusion_model_currulao_v1/variables/variables.data-00000-of-00001")
19
 
20
 
21
  def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: int, temperature: int):
 
15
 
16
  sd_dreambooth_model_1.load_weights_from_pytorch_ckpt("riffusion-model-v1.ckpt")
17
 
18
+ sd_dreambooth_model_1.diffusion_model.load_weights("dreambooth_riffusion_model_currulao_v1/")
19
 
20
 
21
  def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: int, temperature: int):