sagarkariya commited on
Commit
8d4bf37
Β·
verified Β·
1 Parent(s): 8230759

added token use_auth_token=hf_token

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ else:
18
  # 🎯 Load CodeLlama 3B Model
19
  model_name = "codellama/CodeLlama-3b"
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
- model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
22
 
23
  # πŸ” Function to Generate Responses
24
  def generate_response(prompt):
 
18
  # 🎯 Load CodeLlama 3B Model
19
  model_name = "codellama/CodeLlama-3b"
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
+ model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto",use_auth_token=hf_token)
22
 
23
  # πŸ” Function to Generate Responses
24
  def generate_response(prompt):