maomao88 commited on
Commit
4071b28
·
1 Parent(s): 24b8880

update transformers version in requirements.txt

Browse files
.DS_Store ADDED
Binary file (6.15 kB). View file
 
backend/hf_model_utils.py CHANGED
@@ -17,7 +17,6 @@ from transformers import (
17
  )
18
  from accelerate import init_empty_weights
19
 
20
-
21
  model_structure_cache = {}
22
 
23
 
@@ -85,7 +84,6 @@ def get_model_structure(model_name: str, model_type: str | None):
85
  if model_name in model_structure_cache:
86
  return model_structure_cache[model_name]
87
 
88
- print(model_type)
89
  # 2. If not cached, build the structure
90
  if model_type == "causal":
91
  config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
 
17
  )
18
  from accelerate import init_empty_weights
19
 
 
20
  model_structure_cache = {}
21
 
22
 
 
84
  if model_name in model_structure_cache:
85
  return model_structure_cache[model_name]
86
 
 
87
  # 2. If not cached, build the structure
88
  if model_type == "causal":
89
  config = AutoConfig.from_pretrained(model_name, trust_remote_code=True)
backend/requirements.txt CHANGED
@@ -31,7 +31,7 @@ sympy==1.14.0
31
  tokenizers==0.22.0
32
  torch==2.8.0
33
  tqdm==4.67.1
34
- transformers>=4.57.0
35
  typing-inspection==0.4.1
36
  typing_extensions==4.15.0
37
  urllib3==2.5.0
 
31
  tokenizers==0.22.0
32
  torch==2.8.0
33
  tqdm==4.67.1
34
+ transformers>= 5.12.0
35
  typing-inspection==0.4.1
36
  typing_extensions==4.15.0
37
  urllib3==2.5.0