aiqtech commited on
Commit
76ce631
·
verified ·
1 Parent(s): 64e25d4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ def initialize_models():
38
 
39
  try:
40
  # 번역 모델 - 가벼운 버전 사용
41
- model = AutoModelForSeq2SeqLength.from_pretrained(
42
  model_name,
43
  low_cpu_mem_usage=True,
44
  torch_dtype=torch.float16
 
38
 
39
  try:
40
  # 번역 모델 - 가벼운 버전 사용
41
+ model = AutoModelForSeq2SeqLM.from_pretrained( # 수정된 부분
42
  model_name,
43
  low_cpu_mem_usage=True,
44
  torch_dtype=torch.float16