Spaces:
Running
on
Zero
Running
on
Zero
Update app_inpaint2.py
Browse files- app_inpaint2.py +1 -2
app_inpaint2.py
CHANGED
|
@@ -83,8 +83,7 @@ def fill_image(prompt, image, model_selection, paste_back):
|
|
| 83 |
target_model,
|
| 84 |
torch_dtype=torch.float16,
|
| 85 |
vae=vae,
|
| 86 |
-
controlnet=model
|
| 87 |
-
variant="fp16",
|
| 88 |
)
|
| 89 |
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|
| 90 |
pipe.to("cuda")
|
|
|
|
| 83 |
target_model,
|
| 84 |
torch_dtype=torch.float16,
|
| 85 |
vae=vae,
|
| 86 |
+
controlnet=model
|
|
|
|
| 87 |
)
|
| 88 |
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|
| 89 |
pipe.to("cuda")
|