G-AshwinKumar
commited on
Commit
•
dd4ea0a
1
Parent(s):
b2befb1
update model name in examples
Browse files
README.md
CHANGED
@@ -137,7 +137,7 @@ Use the code below to get started with the model. You can run conversational inf
|
|
137 |
import transformers
|
138 |
import torch
|
139 |
|
140 |
-
model_id = "HPAI-BSC/
|
141 |
|
142 |
pipeline = transformers.pipeline(
|
143 |
"text-generation",
|
@@ -179,7 +179,7 @@ print(outputs[0]["generated_text"][len(prompt):])
|
|
179 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
180 |
import torch
|
181 |
|
182 |
-
model_id = "HPAI-BSC/
|
183 |
|
184 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
185 |
model = AutoModelForCausalLM.from_pretrained(
|
|
|
137 |
import transformers
|
138 |
import torch
|
139 |
|
140 |
+
model_id = "HPAI-BSC/Llama3.1-Aloe-Beta-8B"
|
141 |
|
142 |
pipeline = transformers.pipeline(
|
143 |
"text-generation",
|
|
|
179 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
180 |
import torch
|
181 |
|
182 |
+
model_id = "HPAI-BSC/Llama3.1-Aloe-Beta-8B"
|
183 |
|
184 |
tokenizer = AutoTokenizer.from_pretrained(model_id)
|
185 |
model = AutoModelForCausalLM.from_pretrained(
|