Spaces:
Running
Running
Commit
·
494b13c
1
Parent(s):
3a19f67
Used token instead of use_auth_token
Browse files- 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 |
+
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 |
)
|