Update app.py

app.py

@@ -63,9 +63,9 @@ from diffusers import DiffusionPipeline
 # FLUX 모델 설정
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if torch.cuda.is_available() else torch.float32
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16)
+pipe.enable_model_cpu_offload()
 
-pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=dtype)
-pipe = pipe.to(device)
 
 # 메모리 최적화 (GPU 사용 시에만 적용)
 if torch.cuda.is_available():