Update app.py

app.py

@@ -79,11 +79,12 @@ gd_model = GroundingDinoForObjectDetection.from_pretrained(gd_model_path, torch_
 gd_model = gd_model.to(device=device)
 assert isinstance(gd_model, GroundingDinoForObjectDetection)
 
+# FLUX 파이프라인 초기화 부분 수정
 pipe = FluxPipeline.from_pretrained(
     "black-forest-labs/FLUX.1-dev",
     torch_dtype=torch.float16,  # 메모리 사용량 감소를 위해 float16 사용
     use_auth_token=HF_TOKEN,
-    device_map="auto"  # 자동 장치 매핑
+    device_map="balanced"  # 'auto' 대신 'balanced' 사용
 )
 pipe.load_lora_weights(
     hf_hub_download(
@@ -95,6 +96,8 @@ pipe.load_lora_weights(
 pipe.fuse_lora(lora_scale=0.125)
 
 
+
+
 class timer:
     def __init__(self, method_name="timed process"):
         self.method = method_name