Spaces:
Sleeping
Sleeping
arjunanand13
commited on
Commit
•
1e55ce6
1
Parent(s):
8c96f84
Update app.py
Browse files
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|>")
|