Spaces:
Running
Running
update transformers version in requirements.txt
Browse files- .DS_Store +0 -0
- backend/hf_model_utils.py +0 -2
- backend/requirements.txt +1 -1
.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>=
|
| 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
|