Text Generation
Transformers
Safetensors
Chinese
English
qwen
conversational
custom_code
yuyijiong commited on
Commit
db25d91
1 Parent(s): 0ef8b95

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -0
README.md CHANGED
@@ -31,6 +31,7 @@ pipeline_tag: text-generation
31
  ```python
32
  from transformers import AutoModelForCausalLM, AutoTokenizer,AutoConfig
33
 
 
34
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True,use_fast=False)
35
  model = AutoModelForCausalLM.from_pretrained(model_path, device_map="auto", trust_remote_code=True,load_in_4bit=True).eval()
36
 
 
31
  ```python
32
  from transformers import AutoModelForCausalLM, AutoTokenizer,AutoConfig
33
 
34
+ model_path="yuyijiong/Qwen-14b-chat-yarn-32k"
35
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True,use_fast=False)
36
  model = AutoModelForCausalLM.from_pretrained(model_path, device_map="auto", trust_remote_code=True,load_in_4bit=True).eval()
37