arjunanand13 commited on
Commit
1e55ce6
1 Parent(s): 8c96f84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ bnb_config = transformers.BitsAndBytesConfig(
41
  )
42
 
43
  tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct",token=HF_TOKEN)
44
- model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto",token=HF_TOKEN) # to("cuda:0")
45
  terminators = [
46
  tokenizer.eos_token_id,
47
  tokenizer.convert_tokens_to_ids("<|eot_id|>")
 
41
  )
42
 
43
  tokenizer = AutoTokenizer.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct",token=HF_TOKEN)
44
+ model = AutoModelForCausalLM.from_pretrained("meta-llama/Meta-Llama-3-8B-Instruct", device_map="auto",token=HF_TOKEN,quantization_config=bnb_config) # to("cuda:0")
45
  terminators = [
46
  tokenizer.eos_token_id,
47
  tokenizer.convert_tokens_to_ids("<|eot_id|>")