Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,7 @@ vae = AutoencoderKL.from_pretrained(model_repo_id, subfolder="vae", torch_dtype=
|
|
| 26 |
unet = UNet2DConditionModel.from_pretrained(model_repo_id, subfolder="unet", torch_dtype=torch_dtype, allow_pickle=False)
|
| 27 |
text_encoder = CLIPTextModel.from_pretrained(model_repo_id, subfolder="text_encoder", torch_dtype=torch_dtype)
|
| 28 |
tokenizer = CLIPTokenizer.from_pretrained(model_repo_id, subfolder="tokenizer", torch_dtype=torch_dtype)
|
| 29 |
-
pipe =
|
| 30 |
|
| 31 |
|
| 32 |
try:
|
|
|
|
| 26 |
unet = UNet2DConditionModel.from_pretrained(model_repo_id, subfolder="unet", torch_dtype=torch_dtype, allow_pickle=False)
|
| 27 |
text_encoder = CLIPTextModel.from_pretrained(model_repo_id, subfolder="text_encoder", torch_dtype=torch_dtype)
|
| 28 |
tokenizer = CLIPTokenizer.from_pretrained(model_repo_id, subfolder="tokenizer", torch_dtype=torch_dtype)
|
| 29 |
+
pipe = MarigoldPipeline(vae=vae, unet=unet, text_encoder=text_encoder, tokenizer=tokenizer)
|
| 30 |
|
| 31 |
|
| 32 |
try:
|