Update app.py

app.py

@@ -68,20 +68,19 @@ densepose_predictor = DensePosePredictor(
     config_path="./ckpts/densepose/densepose_rcnn_R_50_FPN_s1x.yaml",
     weights_path="./ckpts/densepose/model_final_162be9.pkl",
 )
-
-# Leffa 모델 초기화
+# Leffa 모델 초기화 수정
 vt_model = LeffaModel(
     pretrained_model_name_or_path="./ckpts/stable-diffusion-inpainting",
-    pretrained_model="./ckpts/virtual_tryon.pth",
-    device_map="auto"
+    pretrained_model="./ckpts/virtual_tryon.pth"
 )
+vt_model.to(device)  # 모델을 GPU로 이동
 vt_inference = LeffaInference(model=vt_model)
 
 pt_model = LeffaModel(
     pretrained_model_name_or_path="./ckpts/stable-diffusion-xl-1.0-inpainting-0.1",
-    pretrained_model="./ckpts/pose_transfer.pth",
-    device_map="auto"
+    pretrained_model="./ckpts/pose_transfer.pth"
 )
+pt_model.to(device)  # 모델을 GPU로 이동
 pt_inference = LeffaInference(model=pt_model)