Spaces:

arjunanand13 commited on
Commit
f623b49
1 Parent(s): f57c40f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ device = f'cuda:{cuda.current_device()}' if cuda.is_available() else 'cpu'
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,
45
  tokenizer.convert_tokens_to_ids("<|eot_id|>")
 
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",token=HF_TOKEN) # to("cuda:0")
43
  terminators = [
44
  tokenizer.eos_token_id,
45
  tokenizer.convert_tokens_to_ids("<|eot_id|>")