Spaces:
Running
Running
Commit
·
3514b7e
1
Parent(s):
9b00313
Revert "Used token instead of use_auth_token"
Browse filesThis reverts commit 494b13c2fe11554d828631f99562cde3f81b3c35.
- model/analyzer.py +1 -1
model/analyzer.py
CHANGED
@@ -25,7 +25,7 @@ def analyze_script(script):
|
|
25 |
# Load model with token authentication
|
26 |
model = AutoModelForCausalLM.from_pretrained(
|
27 |
"meta-llama/Llama-3.2-1B",
|
28 |
-
|
29 |
torch_dtype=torch.float16 if device == "cuda" else torch.float32, # Use 16-bit precision for CUDA, 32-bit for CPU
|
30 |
device_map="auto" # Automatically map model to available device
|
31 |
)
|
|
|
25 |
# Load model with token authentication
|
26 |
model = AutoModelForCausalLM.from_pretrained(
|
27 |
"meta-llama/Llama-3.2-1B",
|
28 |
+
use_auth_token=hf_token, # Pass the token to authenticate
|
29 |
torch_dtype=torch.float16 if device == "cuda" else torch.float32, # Use 16-bit precision for CUDA, 32-bit for CPU
|
30 |
device_map="auto" # Automatically map model to available device
|
31 |
)
|