Update app.py
Browse files
app.py
CHANGED
|
@@ -55,6 +55,7 @@ print("Loading SD2 Inpainting...")
|
|
| 55 |
inpaint = StableDiffusionInpaintPipeline.from_pretrained(
|
| 56 |
"sd2-community/stable-diffusion-2-inpainting",
|
| 57 |
torch_dtype=torch.float16,
|
|
|
|
| 58 |
).to("cuda")
|
| 59 |
|
| 60 |
print("SD2 Inpainting ready.")
|
|
@@ -81,6 +82,7 @@ blip_processor = BlipProcessor.from_pretrained("Salesforce/blip-image-captioning
|
|
| 81 |
blip_model = BlipForConditionalGeneration.from_pretrained(
|
| 82 |
"Salesforce/blip-image-captioning-base",
|
| 83 |
torch_dtype=torch.float16,
|
|
|
|
| 84 |
).to("cuda")
|
| 85 |
print("All models loaded.")
|
| 86 |
|
|
|
|
| 55 |
inpaint = StableDiffusionInpaintPipeline.from_pretrained(
|
| 56 |
"sd2-community/stable-diffusion-2-inpainting",
|
| 57 |
torch_dtype=torch.float16,
|
| 58 |
+
resume_download=True
|
| 59 |
).to("cuda")
|
| 60 |
|
| 61 |
print("SD2 Inpainting ready.")
|
|
|
|
| 82 |
blip_model = BlipForConditionalGeneration.from_pretrained(
|
| 83 |
"Salesforce/blip-image-captioning-base",
|
| 84 |
torch_dtype=torch.float16,
|
| 85 |
+
resume_download= True
|
| 86 |
).to("cuda")
|
| 87 |
print("All models loaded.")
|
| 88 |
|