Update scripts/mainrunpodA1111.py

scripts/mainrunpodA1111.py

@@ -238,7 +238,7 @@ def modeldwn(model_LINK):
             modelname=get_true_name(model_LINK)
             loramodel=f'/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/{modelname}'
             if not os.path.exists(loramodel):
-              gdown.download(url=LoRA_LINK.replace('/file/d/', '/uc?id=').replace('/view', ''), output=loramodel, quiet=False)
+              gdown.download(url=model_LINK.replace('/file/d/', '/uc?id=').replace('/view', ''), output=loramodel, quiet=False)
               clear_output()
             else:
               print('Model already exists')
@@ -498,7 +498,7 @@ def getsrc(url):
     return src
 
 def get_true_name(url):
-    file_id = url.split('/')[-2]
+    file_id = url.split('/')[-1]
     gdrive_url = f'https://drive.google.com/uc?id={file_id}'
     return gdown.download(url=gdrive_url, quiet=True)