Spaces:
Sleeping
Sleeping
fix model_name
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ login(token)
|
|
10 |
|
11 |
model_name = 'meta-llama/Meta-Llama-3.1-8B-Instruct'
|
12 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype = torch.bfloat16)
|
13 |
-
tokenizer = AutoTokenizer.from_pretrained(
|
14 |
|
15 |
if torch.cuda.is_available():
|
16 |
device = torch.device('cuda')
|
|
|
10 |
|
11 |
model_name = 'meta-llama/Meta-Llama-3.1-8B-Instruct'
|
12 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype = torch.bfloat16)
|
13 |
+
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
14 |
|
15 |
if torch.cuda.is_available():
|
16 |
device = torch.device('cuda')
|