juancopi81 commited on
Commit
9aa89b6
1 Parent(s): 9f97e26

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ sd_dreambooth_model_1=StableDiffusionPy(resolution, resolution, download_weights
10
 
11
  sd_dreambooth_model_1.load_weights_from_pytorch_ckpt("riffusion-model-v1.ckpt")
12
 
13
- sd_dreambooth_model_1.diffusion_model.load_weights("dreambooth_riffusion_model_currulao_v1/variables")
14
 
15
 
16
  def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: int, temperature: int):
 
10
 
11
  sd_dreambooth_model_1.load_weights_from_pytorch_ckpt("riffusion-model-v1.ckpt")
12
 
13
+ #sd_dreambooth_model_1.diffusion_model.load_weights("dreambooth_riffusion_model_currulao_v1/variables")
14
 
15
 
16
  def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: int, temperature: int):