Update app.py
Browse files
app.py
CHANGED
@@ -45,7 +45,7 @@ def infer(
|
|
45 |
|
46 |
pipe = None
|
47 |
if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"):
|
48 |
-
pipe=DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype,cross_attention_kwargs={"scale":
|
49 |
pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
|
50 |
else:
|
51 |
print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
|
@@ -99,7 +99,7 @@ with gr.Blocks(css=css) as demo:
|
|
99 |
minimum=0,
|
100 |
maximum=2,
|
101 |
step=0.05,
|
102 |
-
value=1,
|
103 |
)
|
104 |
|
105 |
|
|
|
45 |
|
46 |
pipe = None
|
47 |
if (model_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"):
|
48 |
+
pipe=DiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype,cross_attention_kwargs={"scale": lscale})
|
49 |
pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
|
50 |
else:
|
51 |
print("stable-diffusion-v1-5/stable-diffusion-v1-5 with lora")
|
|
|
99 |
minimum=0,
|
100 |
maximum=2,
|
101 |
step=0.05,
|
102 |
+
value=1,
|
103 |
)
|
104 |
|
105 |
|