Upload 4 files

a1111_drive_link.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+"""a1111_drive_link.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/#fileId=https%3A//huggingface.co/arcacolab/drive_link/blob/main/a1111_drive_link.ipynb
+
+<a href="https://colab.research.google.com/github/arcacolab/oneclick/blob/main/a1111_drive_link.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
+"""
+
+import os
+
+# 모델 폴더 생성
+dir_path = "/content/A1111/models/Stable-diffusion/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/A1111/models/Lora/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/A1111/models/VAE/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/A1111/embeddings"
+os.makedirs(dir_path, exist_ok=True)
+
+# Google Drive 경로 설정
+sdpath = "/content/drive/MyDrive/ONECLICK/"
+outputspath = sdpath + "outputs"
+modelpath = sdpath + "model"
+lorapath = sdpath + "lora"
+vaepath = sdpath + "vae"
+embeddingspath = sdpath + "embedding"
+
+# outputs 폴더 생성 및 심볼릭 링크
+if not os.path.exists(outputspath):
+  !mkdir -p -v "{outputspath}"
+
+link_path = "/content/A1111/outputs"
+if not os.path.exists(link_path) and not os.path.islink(link_path):
+    !ln -s -v "{outputspath}" "{link_path}"
+
+# model 폴더 생성 및 심볼릭 링크
+if not os.path.exists(modelpath):
+  !mkdir -p -v "{modelpath}"
+!ln -s -v "{modelpath}" "/content/A1111/models/Stable-diffusion"
+
+# lora 폴더 생성 및 심볼릭 링크
+if not os.path.exists(lorapath):
+  !mkdir -p -v "{lorapath}"
+!ln -s -v "{lorapath}" "/content/A1111/models/Lora"
+
+# vae 폴더 생성 및 심볼릭 링크
+if not os.path.exists(vaepath):
+  !mkdir -p -v "{vaepath}"
+!ln -s -v "{vaepath}" "/content/A1111/models/VAE"
+
+# embeddings 폴더 생성 및 심볼릭 링크
+if not os.path.exists(embeddingspath):
+  !mkdir -p -v "{embeddingspath}"
+!ln -s -v "{embeddingspath}" "/content/A1111/embeddings"

comfyui_drive_link.py

@@ -0,0 +1,64 @@
+# -*- coding: utf-8 -*-
+"""comfyui_drive_link.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/#fileId=https%3A//huggingface.co/arcacolab/drive_link/blob/main/comfyui_drive_link.ipynb
+
+<a href="https://colab.research.google.com/github/arcacolab/oneclick/blob/main/comfyui_drive_link.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
+"""
+
+import os
+source_checkpoints = "/content/ComfyUI/models/checkpoints"
+target_checkpoints = "/content/drive/MyDrive/ONECLICK/model"
+source_loras = "/content/ComfyUI/models/loras"
+target_loras = "/content/drive/MyDrive/ONECLICK/lora"
+source_vae = "/content/ComfyUI/models/vae"
+target_vae = "/content/drive/MyDrive/ONECLICK/vae"
+source_embeddings = "/content/ComfyUI/models/embeddings"
+target_embeddings = "/content/drive/MyDrive/ONECLICK/embedding"
+
+os.makedirs(target_checkpoints, exist_ok=True)
+os.makedirs(target_loras, exist_ok=True)
+os.makedirs(target_vae, exist_ok=True)
+os.makedirs(target_embeddings, exist_ok=True)
+
+!ln -sf "{target_checkpoints}" "{source_checkpoints}"
+!ln -sf "{target_loras}" "{source_loras}"
+!ln -sf "{target_vae}" "{source_vae}"
+!ln -sf "{target_embeddings}" "{source_embeddings}"
+
+#출력물 경로 관련
+import shutil
+
+def delete_output_folder_if_file_exists():
+  """
+  /content/ComfyUI/output 폴더 내부에 _output_images_will_be_put_here 파일이 있는 경우 폴더를 삭제합니다.
+  파일이 없는 경우 폴더를 유지합니다.
+  """
+  folder_path = "/content/ComfyUI/output"
+  file_path = os.path.join(folder_path, "_output_images_will_be_put_here")
+
+  if os.path.exists(file_path):
+    shutil.rmtree(folder_path, ignore_errors=True)
+    print(f"{folder_path} 폴더를 삭제했습니다. (_output_images_will_be_put_here 파일이 존재합니다.)")
+  else:
+    print(f"{folder_path} 폴더를 유지합니다. (_output_images_will_be_put_here 파일이 존재하지 않습니다.)")
+
+# 함수 호출
+delete_output_folder_if_file_exists()
+
+
+import os
+sdpath = "/content/drive/MyDrive/ONECLICK/"
+
+outputspath = sdpath + "output"
+
+
+if not os.path.exists(outputspath):
+    !mkdir -p -v "{outputspath}"
+
+link_path = "/content/ComfyUI/output"
+if not os.path.exists(link_path) and not os.path.islink(link_path):
+    !ln -s -v "{outputspath}" "{link_path}"

forge_drive_link.py

@@ -0,0 +1,58 @@
+# -*- coding: utf-8 -*-
+"""forge_drive_link.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/#fileId=https%3A//huggingface.co/arcacolab/drive_link/blob/main/forge_drive_link.ipynb
+
+<a href="https://colab.research.google.com/github/arcacolab/oneclick/blob/main/forge_drive_link.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
+"""
+
+import os
+
+# 모델 폴더 생성
+dir_path = "/content/Forge/models/Stable-diffusion/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/Forge/models/Lora/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/Forge/models/VAE/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/Forge/embeddings"
+os.makedirs(dir_path, exist_ok=True)
+
+# Google Drive 경로 설정
+sdpath = "/content/drive/MyDrive/ONECLICK/"
+outputspath = sdpath + "outputs"
+modelpath = sdpath + "model"
+lorapath = sdpath + "lora"
+vaepath = sdpath + "vae"
+embeddingspath = sdpath + "embedding"
+
+# outputs 폴더 생성 및 심볼릭 링크
+if not os.path.exists(outputspath):
+  !mkdir -p -v "{outputspath}"
+
+link_path = "/content/Forge/outputs"
+if not os.path.exists(link_path) and not os.path.islink(link_path):
+    !ln -s -v "{outputspath}" "{link_path}"
+
+# model 폴더 생성 및 심볼릭 링크
+if not os.path.exists(modelpath):
+  !mkdir -p -v "{modelpath}"
+!ln -s -v "{modelpath}" "/content/Forge/models/Stable-diffusion"
+
+# lora 폴더 생성 및 심볼릭 링크
+if not os.path.exists(lorapath):
+  !mkdir -p -v "{lorapath}"
+!ln -s -v "{lorapath}" "/content/Forge/models/Lora"
+
+# vae 폴더 생성 및 심볼릭 링크
+if not os.path.exists(vaepath):
+  !mkdir -p -v "{vaepath}"
+!ln -s -v "{vaepath}" "/content/Forge/models/VAE"
+
+# embeddings 폴더 생성 및 그시기
+if not os.path.exists(embeddingspath):
+  !mkdir -p -v "{embeddingspath}"
+!ln -s -v "{embeddingspath}" "/content/Forge/embeddings"

reforge_drive_link.py

@@ -0,0 +1,59 @@
+# -*- coding: utf-8 -*-
+"""reforge_drive_link.ipynb
+
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/#fileId=https%3A//huggingface.co/arcacolab/drive_link/blob/main/reforge_drive_link.ipynb
+
+<a href="https://colab.research.google.com/github/arcacolab/oneclick/blob/main/reforge_drive_link.ipynb" target="_parent"><img src="https://colab.research.google.com/assets/colab-badge.svg" alt="Open In Colab"/></a>
+"""
+
+import os
+
+# 모델 폴더 생성
+dir_path = "/content/ReForge/models/Stable-diffusion/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/ReForge/models/Lora/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/ReForge/models/VAE/"
+os.makedirs(dir_path, exist_ok=True)
+dir_path = "/content/ReForge/embeddings/"
+os.makedirs(dir_path, exist_ok=True)
+
+
+# Google Drive 경로 설정
+sdpath = "/content/drive/MyDrive/ONECLICK/"
+outputspath = sdpath + "outputs"
+modelpath = sdpath + "model"
+lorapath = sdpath + "lora"
+vaepath = sdpath + "vae"
+embeddingspath = sdpath + "embedding"
+
+# outputs 폴더 생성 및 심볼릭 링크
+if not os.path.exists(outputspath):
+  !mkdir -p -v "{outputspath}"
+
+link_path = "/content/ReForge/outputs"
+if not os.path.exists(link_path) and not os.path.islink(link_path):
+    !ln -s -v "{outputspath}" "{link_path}"
+
+# model 폴더 생성 및 심볼릭 링크
+if not os.path.exists(modelpath):
+  !mkdir -p -v "{modelpath}"
+!ln -s -v "{modelpath}" "/content/ReForge/models/Stable-diffusion"
+
+# lora 폴더 생성 및 심볼릭 링크
+if not os.path.exists(lorapath):
+  !mkdir -p -v "{lorapath}"
+!ln -s -v "{lorapath}" "/content/ReForge/models/Lora"
+
+# vae 폴더 생성 및 심볼릭 링크
+if not os.path.exists(vaepath):
+  !mkdir -p -v "{vaepath}"
+!ln -s -v "{vaepath}" "/content/ReForge/models/VAE"
+
+# embeddings 폴더 생성 및 그시기
+if not os.path.exists(embeddingspath):
+  !mkdir -p -v "{embeddingspath}"
+!ln -s -v "{embeddingspath}" "/content/ReForge/embeddings"