Update app.py

app.py

@@ -26,10 +26,11 @@ model_path = snapshot_download(
 
 # 모델 로드 함수
 def load_pipeline(pipeline_type):
+    logging.debug(f'Loading pipeline: {pipeline_type}')
     if pipeline_type == "text2img":
-        return StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+        return StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16, use_fast=True)
     elif pipeline_type == "img2img":
-        return StableDiffusion3Img2ImgPipeline.from_pretrained(model_path, torch_dtype=torch.float16)
+        return StableDiffusion3Img2ImgPipeline.from_pretrained(model_path, torch_dtype=torch.float16, use_fast=True)
 
 # 디바이스 설정
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")