Hugging Face's logo

BAAI
/
Aquila-7B

Transformers
PyTorch
aquila
custom_code
Model card Files
xet
 Community
1
shunxing1234 commited on
Commit
8691937
·
1 Parent(s): e77ef02

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -40,8 +40,8 @@ from transformers import AutoTokenizer, AutoModelForCausalLM
40
  import torch
41
  from cyg_conversation import covert_prompt_to_input_ids_with_history
42
 
43
- tokenizer = AutoTokenizer.from_pretrained("BAAI/AquilaChat-7B")
44
- model = AutoModelForCausalLM.from_pretrained("BAAI/AquilaChat-7B")
45
  model.eval()
46
  model.to("cuda:0")
47
  vocab = tokenizer.vocab
 
40
  import torch
41
  from cyg_conversation import covert_prompt_to_input_ids_with_history
42
 
43
+ tokenizer = AutoTokenizer.from_pretrained("BAAI/Aquila-7B")
44
+ model = AutoModelForCausalLM.from_pretrained("BAAI/Aquila-7B")
45
  model.eval()
46
  model.to("cuda:0")
47
  vocab = tokenizer.vocab