본문 바로가기
Data & Programming/ChatGPT

[ChatGPT] Stable Diffusion 로라(LoRA) 사용법

by 눈부셔™ 2023. 7. 9.
728x90
 [ChatGPT] Stable Diffusion 로라(LoRA) 사용법

로라(LoRA)는 기존의 모델에 얼굴, 머리, 옷, 배경 등 어느 한 부분만 추가적으로 그릴 수 있도록 학습된 모델로, Stable Diffusion의 체크 포인트 모델 성능을 향상시킵니다.

 

01. https://civitai.com/ 에 접속해서 LoRA를 다운 받습니다.

 

02. 다운로드 하여, C:\stable-diffusion-webui\models\Lora 에 붙여 넣습니다.

 

03. 기본모델이 되는 부분은 majicMIX realistic 을 클릭하여 다운받아,

      C:\stable-diffusion-webui\models\Stable-diffusion 폴더안에 복사해둔다.

 

04. 체크포인트를 새로고침해서, 해당 모델을 선택하고 이미지를 업로드 한다.

 

05. 생성버튼을 누르면이미지가 생성되는데, 전 메모리 부족으로 50%정도 생성되다가 멈추네요. 

OutOfMemoryError: CUDA out of memory. Tried to allocate 208.00 MiB (GPU 0; 4.00 GiB total capacity; 3.07 GiB already allocated; 0 bytes free; 3.28 GiB reserved in total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF

소요 시간 : 24.34s

Torch active/reserved: 3279/3428 MiB, Sys VRAM: 4096/4096 MiB (100.0%)

 

06. 메모리 사용을 줄이기 위해, 고해상도 보정 체크를 지우고 다시 생성해봅니다.

728x90