Spaces:
Runtime error
Runtime error
Commit
•
a4f9fc2
1
Parent(s):
f3d8154
Update app.py
Browse files
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/
|
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):
|