Update app.py

app.py

@@ -5,7 +5,7 @@ import uuid
 import torch
 import subprocess
 from huggingface_hub import snapshot_download
-from diffusers import StableDiffusion3Pipeline, StableDiffusion3Img2ImgPipeline
+from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline
 from transformers import pipeline
 
 # 로깅 설정
@@ -19,9 +19,7 @@ intents.message_content = True
 huggingface_token = os.getenv("HF_TOKEN")
 model_path = snapshot_download(
     repo_id="Corcelio/mobius",
-    revision="refs/pr/26",
     repo_type="model",
-    ignore_patterns=[".md", "..gitattributes"],
     local_dir="mobius",
     token=huggingface_token,
 )
@@ -30,9 +28,9 @@ model_path = snapshot_download(
 def load_pipeline(pipeline_type):
     logging.debug(f'파이프라인 로드 중: {pipeline_type}')
     if pipeline_type == "text2img":
-        return StableDiffusion3Pipeline.from_pretrained(model_path, torch_dtype=torch.float16, use_fast=True)
+        return StableDiffusionPipeline.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, use_fast=True)
+        return StableDiffusionImg2ImgPipeline.from_pretrained(model_path, torch_dtype=torch.float16, use_fast=True)
 
 # 디바이스 설정
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")