Update app.py
Browse files
app.py
CHANGED
@@ -14,6 +14,7 @@ if torch.cuda.is_available():
|
|
14 |
else:
|
15 |
torch_dtype = torch.float32
|
16 |
|
|
|
17 |
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
18 |
#pipe.unet.load_adapter("um235/cartoon_cat_stickers")
|
19 |
pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
|
|
|
14 |
else:
|
15 |
torch_dtype = torch.float32
|
16 |
|
17 |
+
print("initialize", model_repo_id)
|
18 |
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
19 |
#pipe.unet.load_adapter("um235/cartoon_cat_stickers")
|
20 |
pipe.unet = PeftModel.from_pretrained(pipe.unet,"um235/cartoon_cat_stickers")
|