Update models/util.py

models/util.py

@@ -420,10 +420,16 @@ def load_flow_model(
         #     print_load_warning(missing, unexpected)
     return model
 
-
-def load_t5(device: str | torch.device = "cuda", max_length: int = 512) -> HFEmbedder:
-    # max length 64, 128, 256 and 512 should work (if your sequence is short enough)
-    return HFEmbedder("google/t5-v1_1-xxl", max_length=max_length, torch_dtype=torch.bfloat16).to(device)
+# 약 426번 줄에 위치한 load_t5 함수를 찾아 다음과 같이 수정
+def load_t5(device, max_length=256):
+    try:
+        # 원래 코드: 대형 T5-XXL 모델 로드 시도
+        return HFEmbedder("google/t5-v1_1-xxl", max_length=max_length, torch_dtype=torch.bfloat16).to(device)
+    except Exception as e:
+        print(f"T5-XXL 모델 로딩 실패: {str(e)}")
+        print("더 작은 T5 모델로 대체합니다...")
+        # 더 작은 모델로 대체
+        return HFEmbedder("google/t5-v1_1-large", max_length=max_length, torch_dtype=torch.bfloat16).to(device)
 
 
 def load_clip(device: str | torch.device = "cuda") -> HFEmbedder: