gdown error fixed

2023-12-16 08:36:57 +09:00
parent 42761067ca
commit 6a723cac97
2 changed files with 4 additions and 2 deletions
@@ -229,8 +229,10 @@ class S3Bucket_Load_LoRA:
            
            local_lora_path = str(local_lora_path)
            if "drive.google" in lora_url_or_path:
-                import gdown
-                gdown.download(lora_url_or_path, local_lora_path, quiet=False)
+                local_lora_path = download_file_from_url(url=lora_url_or_path, download_path=local_lora_path)
+
+                # import gdown
+                # gdown.download(lora_url_or_path, local_lora_path, quiet=False)
            else:
                local_lora_path = download_file_from_s3_bucket(bucket_file_path=lora_url_or_path, download_path=local_lora_path)