
490
2
부
모델 사용
EulerAncestralDiscreteScheduler,
)
그다음
model
_
id
를 설정하고 적절한 버전의 스케줄러를 선택합니다.
model_id = "stabilityai/stable-diffusion-2"
# 장치를 선택합니다.
device = "cuda" if torch.cuda.is_available() else "cpu"
# 1. 스케줄러를 선택합니다.
scheduler = EulerAncestralDiscreteScheduler.from_pretrained(
model_id,
subfolder="scheduler"
)
그다음
StableDiffusionPipeline
클래스를 사용해 파이프라인을 만들고 가속기에 로드합니다.
# 2. 선택한 스케줄러로 파이프라인을 로드합니다.
pipe = StableDiffusionPipeline.from_pretrained(
model_id,
scheduler=scheduler,
torch_dtype=torch.float16
).to(device)
다음 단계는 모델의 새로운 동작을 결정하도록 훈련된
LoRA
가중치를 할당하는 것입니다.
# 3. (선택사항) LoRA 가중치를 로드합니다.
pipe.load_lora_weights("lmoroney/finetuned-misato-sd2" ...