Spaces:
Runtime error
Runtime error
arjunanand13
commited on
Commit
•
f623b49
1
Parent(s):
f57c40f
Update app.py
Browse files
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|>")
|