Futuretop commited on
Commit
9c304c0
·
verified ·
1 Parent(s): c4f5519

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -75,21 +75,21 @@ model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
75
  # Florence-2 로드
76
  device = "cuda" if torch.cuda.is_available() else "cpu"
77
  scheduler = EulerDiscreteScheduler.from_pretrained(
78
- model_id,
79
  subfolder="scheduler",
80
  torch_dtype=torch.float16,
81
  )
82
  text_encoder = CLIPTextModel.from_pretrained(
83
- model_id, subfolder="text_encoder_1", torch_dtype=torch.float16
84
  )
85
  tokenizer = CLIPTokenizer.from_pretrained(
86
- model_id, subfolder="tokenizer_1"
87
  )
88
  feature_extractor = CLIPFeatureExtractor.from_pretrained(
89
  model_repo, subfolder="feature_extractor"
90
  )
91
  unet = UNet2DConditionModel.from_pretrained(
92
- model_id, subfolder="unet", torch_dtype=torch.float16
93
  )
94
  florence_model = AutoModelForCausalLM.from_pretrained(LOCAL_FLORENCE, trust_remote_code=True, torch_dtype=torch.float16)
95
  florence_model.to("cpu")
 
75
  # Florence-2 로드
76
  device = "cuda" if torch.cuda.is_available() else "cpu"
77
  scheduler = EulerDiscreteScheduler.from_pretrained(
78
+ model_repo,
79
  subfolder="scheduler",
80
  torch_dtype=torch.float16,
81
  )
82
  text_encoder = CLIPTextModel.from_pretrained(
83
+ model_repo, subfolder="text_encoder_1", torch_dtype=torch.float16
84
  )
85
  tokenizer = CLIPTokenizer.from_pretrained(
86
+ model_repo, subfolder="tokenizer_1"
87
  )
88
  feature_extractor = CLIPFeatureExtractor.from_pretrained(
89
  model_repo, subfolder="feature_extractor"
90
  )
91
  unet = UNet2DConditionModel.from_pretrained(
92
+ model_repo, subfolder="unet", torch_dtype=torch.float16
93
  )
94
  florence_model = AutoModelForCausalLM.from_pretrained(LOCAL_FLORENCE, trust_remote_code=True, torch_dtype=torch.float16)
95
  florence_model.to("cpu")