Spaces:

um235
/

diffusion

Sleeping

um235 commited on Feb 10

Commit

b40ecd3

verified ·

1 Parent(s): 348048e

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 = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
     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 = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
     pipe = pipe.to(device)
     image = pipe(