Kuberwastaken commited on
Commit
494b13c
·
1 Parent(s): 3a19f67

Used token instead of use_auth_token

Browse files
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
- 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
  )
 
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
  )