piuzha commited on
Commit
b182c82
1 Parent(s): df95f7d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ torch.backends.cuda.enable_mem_efficient_sdp(False)
27
  torch.backends.cuda.enable_flash_sdp(False)
28
 
29
 
30
- model_name = 'moxin-org/moxin-7b'
31
  tokenizer = AutoTokenizer.from_pretrained(model_name)
32
  model = AutoModelForCausalLM.from_pretrained(
33
  model_name,
 
27
  torch.backends.cuda.enable_flash_sdp(False)
28
 
29
 
30
+ model_name = 'moxin-org/moxin-llm-7b'
31
  tokenizer = AutoTokenizer.from_pretrained(model_name)
32
  model = AutoModelForCausalLM.from_pretrained(
33
  model_name,