stlaurentjr commited on
Commit
b031058
1 Parent(s): d16623b

Update scripts/mainrunpodA1111.py

Browse files
Files changed (1) hide show
  1. scripts/mainrunpodA1111.py +2 -2
scripts/mainrunpodA1111.py CHANGED
@@ -238,7 +238,7 @@ def modeldwn(model_LINK):
238
  modelname=get_true_name(model_LINK)
239
  loramodel=f'/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}'
240
  if not os.path.exists(loramodel):
241
- gdown.download(url=model_LINK, output=loramodel, quiet=False, fuzzy=True)
242
  clear_output()
243
  else:
244
  print('Model already exists')
@@ -277,7 +277,7 @@ def loradwn(LoRA_LINK):
277
  modelname=get_true_name(LoRA_LINK)
278
  loramodel=f'/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}'
279
  if not os.path.exists(loramodel):
280
- gdown.download(url=LoRA_LINK, output=loramodel, quiet=False, fuzzy=True)
281
  clear_output()
282
  else:
283
  print('Model already exists')
 
238
  modelname=get_true_name(model_LINK)
239
  loramodel=f'/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}'
240
  if not os.path.exists(loramodel):
241
+ gdown.download(url=LoRA_LINK.replace('/file/d/', '/uc?id=').replace('/view', ''), output=loramodel, quiet=False)
242
  clear_output()
243
  else:
244
  print('Model already exists')
 
277
  modelname=get_true_name(LoRA_LINK)
278
  loramodel=f'/workspace/sd/stable-diffusion-webui/models/Lora/{modelname}'
279
  if not os.path.exists(loramodel):
280
+ gdown.download(url=LoRA_LINK.replace('/file/d/', '/uc?id=').replace('/view', ''), output=loramodel, quiet=False)
281
  clear_output()
282
  else:
283
  print('Model already exists')