G-AshwinKumar commited on
Commit
dd4ea0a
1 Parent(s): b2befb1

update model name in examples

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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/Llama31-Aloe-8B-Beta"
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/Llama31-Aloe-Beta-8B"
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(