Spaces:
Sleeping
Sleeping
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|>")
|