AI모델을 통합 관리하는 방법

 

로컬 PC에서 사용할 수 있는 다양한 AI 툴들은 기본적으로 체크포인트, 로라 등의 AI모델들을 사용합니다. 예를 들어 ComfyUI나 webUI Forge에서 사용하는 AI모델들은 동일한 모델들입니다.

 

하지만 각각의 툴마다 고유의 모델 폴더에 AI모델 파일들이 위치해 있어야 합니다. 때문에 여러 가지 AI툴을 사용하는 경우 같은 모델을 여기저기 다 설치해 줘야 합니다. 하드디스크 공간이 낭비되는 것이죠.

 

때문에 이것을 한 폴더에 몰아넣고 통합 관리할 수 있으면 매우 편리할 것입니다.

각각의 Ai툴들은 이렇게 폴더 경로를 변경해 줄 수 있도록 코드를 제공해 줍니다. 하지만 툴마다 그 방법이 다르기도 합니다. 예를 들어 webUI Forge의 경우 비치파일을 열어보면 다음과 같은 문구들이 있을 것입니다.

 

@REM  --ckpt-dir %A1111_HOME%/models/Stable-diffusion ^
@REM  --hypernetwork-dir %A1111_HOME%/models/hypernetworks ^
@REM  --embeddings-dir %A1111_HOME%/embeddings ^
@REM  --lora-dir %A1111_HOME%/models/Lora

 

앞에 @REM은 참조문으로 사용하라는 명령입니다.(해서 정상적인 코드로 인식 못합니다)

각 모델 폴더별로 코드를 활성화하여 사용하라는 안내문인데, 이것을 다음과 같이 해주면 폴더 위치를 바꿔 사용할 수 있습니다.

 

set COMMANDLINE_ARGS=--ckpt-dir "../Models/Stable-diffusion" --lora-dir "../Models/Lora" --embeddings-dir "../Models/embeddings" --controlnet-dir "../Models/ControlNet" --vae-dir "../Models/VAE"

 

이 구문은 각각의 폴더 경로는 한 단계 위의 상위 경로에 있는 폴더로 바꿔 달라는 경로 변경 명령입니다. 이렇게 해두면 CMD에서 포지를 로드할 때 이것을 인식하고 경로변경을 해줍니다.

 

하지만 이와 같은 경우 공동 관리할 모든 모델폴더 등을 변경하기가 쉽지 않습니다. 또한 경로변경 코드가 늘어날수록 애러가 날 확률이 생깁니다. 

 

그래서 통상은 다음과 같은 방법을 사용합니다.

 

디렉터리 심볼릭 링크

mklink /D 명령은 윈도에서 디렉터리 심볼릭 링크(Symbolic Link)를 생성할 때 사용합니다. 쉽게 말해, 실제 폴더는 다른 곳(G 드라이브 등)에 있지만, 마치 특정 위치(C 드라이브 등)에 있는 것처럼 컴퓨터를 속이는 '가짜 연결 통로'를 만드는 것입니다.

 

이것은 바로가기와는 다른 것입니다. 바로가기는 파일로 생성되며 폴더링크 자체가 안됩니다. 공식은 다음과 같습니다.

 

mklink /D "가짜 폴더명" "실제 데이터가 있는 경로"

 

이 명령을 실행하기 위해서는 가짜폴더를 생성할 경로에서 생성해야 합니다.(예 : WebUI_Forge_Toon\webui\models 여기서 가짜폴더 Stable-diffusion를 생성해 줘야 합니다)

 

/D : 디렉터리(폴더) 단위로 링크를 만든다는 옵션입니다.
가짜 폴더명 : 프로그램이 원래 찾으려고 하는 위치 (이 이름의 폴더가 이미 있으면 안 됩니다).
실제 데이터 경로 : 용량이 커서 옮겨둔 실제 모델들이 있는 위치.

 

만드는 방법

1️⃣  먼저 가짜폴더를 생성해야 할 경로(예 : WebUI_Forge_Toon\webui\models)로 이동합니다. 

2️⃣  탐색기의 주소란을 클릭하고 CMD라고 치고 엔터를 누릅니다.

3️⃣  다시 탐색기 주소창에  ..\..\Models\Stable-diffusion 이런 식으로 경로 체크를 해봅니다.

이 경로체크는 넣은 주소 경로에 폴더가 있는지 체크하는 것입니다. 없으면 없다고 애러가 납니다. 

..\ 표시는 현재 경로에서 한 단계 위로 올라가라는 것입니다. 때문에 models 폴더에서 한단계 위로가면 webui폴더가 되고, 또 한단계 더 위로 올라가면 WebUI_Forge_Toon 폴더가 됩니다. 이런 식으로 경로 계산을 해서 ..\ 개수를 채워주면 됩니다.

 

저의 경우를 예로 들겠습니다.

위 그림에서처럼 모델 폴더와 각각의 AI툴들이 동일한 폴더 경로상에 있습니다.

 

실제 모델들이 위치할 경로는 위에 노랑밑줄로 그어진 경로입니다. 2단계를 올려줘야 하죠. 그래서 ..\..\를 넣어주면 됩니다.

해서 CMD에는 mklink /D "Stable-diffusion" "..\..\Models\Stable-diffusion"이렇게 넣고 엔터를 쳐주면 가짜폴더 링크가 생깁니다.

하지만 그전에 반드시 경로 테스트를 위해 탐색기의 주소창에 ..\..\Models\Stable-diffusion라고 넣고 엔터를 쳐봅니다. 

 

체크포인트 모델폴더로 자동 이동하면 경로가 맞는 것입니다. 그럼 다시 뒤로가기를 해서 원래 가짜폴더를 생성해야 할 경로로 온 후에 CMD의 명령을 실행해야 합니다.

 

4️⃣  CMD창에서 

mklink /D "Stable-diffusion" "..\..\Models\Stable-diffusion"
mklink /D "ControlNet" "..\..\Models\ControlNet"
mklink /D "VAE" "..\..\Models\VAE"
mklink /D "Lora" "..\..\Models\Lora"

이런식으로 차례대로 한줄씩 입력하고 엔터를 쳐줍니다. 그러면 위 그림처럼 가상폴더 아이콘이 생성됩니다.

이렇게 해주면 앞으로 공용모델 폴더에 모델들을 넣어주면 AI툴이 알아서 인식해 줍니다.

 

사실 AI툴들은 가짜 폴더를 진짜로 인식 합니다. 다만 윈도우 시스템에서 실제 폴더로 경로를 넘겨주는 것이죠. AI툴의 기능과는 아무 상관이 없고, 다만 윈도우 시스템의 기능인 것입니다. 이 심볼릭 링크파일은 이동식 디스크로 통째로 폴더를 옮겨도 그대로 인식 합니다. 다. 단 폴더위치를 임의대로 바꾸거나 이름을 바꿔 버리면 인식을 하지 못하겠죠.

 

경로생성을 잘 못한 경우

해당 폴더로 가서 이미 생성된 심볼릭링크 파일을 삭제하고, 위의 과정을 다시 수행해주면 됩니다. 안타깝게도 그냥 가짜링크파일을 수정하는 등의 방법은 먹히지 않습니다.