Kuberwastaken commited on
Commit
3514b7e
·
1 Parent(s): 9b00313

Revert "Used token instead of use_auth_token"

Browse files

This reverts commit 494b13c2fe11554d828631f99562cde3f81b3c35.

Files changed (1) hide show
  1. 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
- 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
  )
 
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
  )