Spaces:

um235
/

diffusion

Sleeping

um235 commited on Feb 14

Commit

636d991

verified ·

1 Parent(s): 29ed86a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -50,7 +50,7 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
     pipe = None
-    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"):
         pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
         pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers",cross_attention_kwargs={"scale": 0.5})
     else:

     generator = torch.Generator().manual_seed(seed)
     pipe = None
+    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5 with lora"):
         pipe=DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
         pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers",cross_attention_kwargs={"scale": 0.5})
     else: