Update app.py
Browse files
app.py
CHANGED
@@ -14,12 +14,6 @@ if torch.cuda.is_available():
|
|
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")
|
21 |
-
pipe = pipe.to(device)
|
22 |
-
|
23 |
MAX_SEED = np.iinfo(np.int32).max
|
24 |
MAX_IMAGE_SIZE = 1024
|
25 |
|
|
|
14 |
else:
|
15 |
torch_dtype = torch.float32
|
16 |
|
|
|
|
|
|
|
|
|
|
|
|
|
17 |
MAX_SEED = np.iinfo(np.int32).max
|
18 |
MAX_IMAGE_SIZE = 1024
|
19 |
|