Update scripts/mainrunpodA1111.py

Проверяем, существует ли уже символическая ссылка VAE

scripts/mainrunpodA1111.py

@@ -146,12 +146,25 @@ def mdl(Original_Model_Version, Path_to_MODEL, MODEL_LINK):
     import gdown
     
     src=getsrc(MODEL_LINK)  
-
-    if os.path.exists('/workspace/sd/stable-diffusion-webui/models/VAE'):
-        call('ln -s /workspace/auto-VAE/* /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
-    else:
+    
+    if not os.path.exists('/workspace/sd/stable-diffusion-webui/models/VAE'):
         call('mkdir -p /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
-        call('ln -s /workspace/auto-VAE/* /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
+    
+    # Получаем список файлов в директории /workspace/auto-VAE
+    files = os.listdir('/workspace/auto-VAE')
+    
+    for file in files:
+        source_path = os.path.join('/workspace/auto-VAE', file)
+        target_path = os.path.join('/workspace/sd/stable-diffusion-webui/models/VAE', file)
+        # Проверяем, существует ли уже символическая ссылка
+        if not os.path.exists(target_path):
+            call(f'ln -s {source_path} {target_path}', shell=True)
+
+    #if os.path.exists('/workspace/sd/stable-diffusion-webui/models/VAE'):
+    #    call('ln -s /workspace/auto-VAE/* /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
+    #else:
+    #    call('mkdir -p /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
+    #    call('ln -s /workspace/auto-VAE/* /workspace/sd/stable-diffusion-webui/models/VAE', shell=True)
 
     if not os.path.exists('/workspace/sd/stable-diffusion-webui/models/Stable-diffusion/SDv1-5.ckpt'):
         call('ln -s /workspace/auto-models/* /workspace/sd/stable-diffusion-webui/models/Stable-diffusion', shell=True)