Spaces:

arjunanand13 commited on
Commit
f57c40f
1 Parent(s): 7a542bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -38,7 +38,7 @@ device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
38
  # bnb_4bit_compute_dtype=bfloat16
39
  # )
40
 
41
- tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct")
42
  model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto") # to("cuda:0")
43
  terminators = [
44
  tokenizer.eos_token_id,
 
38
  # bnb_4bit_compute_dtype=bfloat16
39
  # )
40
 
41
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct",token=HF_TOKEN)
42
  model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto") # to("cuda:0")
43
  terminators = [
44
  tokenizer.eos_token_id,