Spaces:

um235
/

diffusion

Sleeping

um235 commited on Feb 10

Commit

3c0798b

verified ·

1 Parent(s): 085f911

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -42,7 +42,7 @@ def infer(
     generator = torch.Generator().manual_seed(seed)
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
-    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"): pipe.unet.load_attn_procs("juliensimon/stable-diffusion-v1-5-pokemon-lora")
     pipe = pipe.to(device)
     image = pipe(

     generator = torch.Generator().manual_seed(seed)
     pipe = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch_dtype)
+    if (model_repo_id=="stable-diffusion-v1-5/stable-diffusion-v1-5"): pipe.unet.load_attn_procs("um235/cartoon_cat_stickers")
     pipe = pipe.to(device)
     image = pipe(