reforge_drive_link.py

# -*- 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"