Madiy commited on
Commit
e35c6e9
·
verified ·
1 Parent(s): 0f14689

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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