OzTianlu commited on
Commit
2ed0063
·
verified ·
1 Parent(s): 21b4315

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -201,12 +201,12 @@ import torch
201
 
202
  # Load model and tokenizer
203
  model = AutoModelForCausalLM.from_pretrained(
204
- "path/to/Asterisk-Pi",
205
  trust_remote_code=True,
206
  torch_dtype=torch.bfloat16,
207
  device_map="auto"
208
  )
209
- tokenizer = AutoTokenizer.from_pretrained("path/to/Asterisk-Pi")
210
 
211
  # Generate text
212
  messages = [{"role": "user", "content": "Explain the waterfall model in software engineering."}]
 
201
 
202
  # Load model and tokenizer
203
  model = AutoModelForCausalLM.from_pretrained(
204
+ "NoesisLab/Asterisk-Pi",
205
  trust_remote_code=True,
206
  torch_dtype=torch.bfloat16,
207
  device_map="auto"
208
  )
209
+ tokenizer = AutoTokenizer.from_pretrained("NoesisLab/Asterisk-Pi")
210
 
211
  # Generate text
212
  messages = [{"role": "user", "content": "Explain the waterfall model in software engineering."}]