Spaces:
Sleeping
Sleeping
laserbeam2045
commited on
Commit
·
a9bf179
1
Parent(s):
1f50e47
fix
Browse files- app.py +5 -3
- requirements.txt +2 -0
app.py
CHANGED
@@ -22,13 +22,15 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
22 |
# -----------------------------------------------------------------------------
|
23 |
tokenizer = AutoTokenizer.from_pretrained(
|
24 |
MODEL_ID,
|
25 |
-
|
26 |
-
trust_remote_code=True
|
|
|
|
|
27 |
)
|
28 |
|
29 |
model = AutoModelForCausalLM.from_pretrained(
|
30 |
MODEL_ID,
|
31 |
-
|
32 |
torch_dtype=torch.float32, # CPU 環境では float32
|
33 |
device_map="auto" if torch.cuda.is_available() else None
|
34 |
)
|
|
|
22 |
# -----------------------------------------------------------------------------
|
23 |
tokenizer = AutoTokenizer.from_pretrained(
|
24 |
MODEL_ID,
|
25 |
+
token=HF_TOKEN,
|
26 |
+
trust_remote_code=True,
|
27 |
+
torch_dtype=torch.float32,
|
28 |
+
low_cpu_mem_usage=True
|
29 |
)
|
30 |
|
31 |
model = AutoModelForCausalLM.from_pretrained(
|
32 |
MODEL_ID,
|
33 |
+
token=HF_TOKEN,
|
34 |
torch_dtype=torch.float32, # CPU 環境では float32
|
35 |
device_map="auto" if torch.cuda.is_available() else None
|
36 |
)
|
requirements.txt
CHANGED
@@ -2,3 +2,5 @@ fastapi
|
|
2 |
uvicorn[standard]
|
3 |
transformers>=4.50.0.dev0
|
4 |
torch
|
|
|
|
|
|
2 |
uvicorn[standard]
|
3 |
transformers>=4.50.0.dev0
|
4 |
torch
|
5 |
+
accelerate>=0.9.0
|
6 |
+
safetensors
|