antjeworring commited on
Commit
5c5e820
·
verified ·
1 Parent(s): f95b7db

fix(readme): body:sub('huggingface\\.co/zenlm/zen4(?![-\\w])')x1;sub('"zenlm/zen4"')x2;sub('zenlm/zen4(?![-\\w])')x1

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -22,12 +22,12 @@ SQL specialist for complex query generation, schema design, query optimization,
22
 
23
  Supports PostgreSQL, MySQL, SQLite, BigQuery, Snowflake, and more.
24
 
25
- Base weights: [zenlm/zen4](https://huggingface.co/zenlm/zen4)
26
 
27
  ```python
28
  from transformers import AutoModelForCausalLM, AutoTokenizer
29
- model = AutoModelForCausalLM.from_pretrained("zenlm/zen4", torch_dtype="auto")
30
- tokenizer = AutoTokenizer.from_pretrained("zenlm/zen4")
31
  messages = [{"role": "user", "content": "Your domain-specific prompt here"}]
32
  text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
33
  inputs = tokenizer(text, return_tensors="pt").to(model.device)
 
22
 
23
  Supports PostgreSQL, MySQL, SQLite, BigQuery, Snowflake, and more.
24
 
25
+ Base weights: [zenlm/zen-pro](https://huggingface.co/zenlm/zen-pro)
26
 
27
  ```python
28
  from transformers import AutoModelForCausalLM, AutoTokenizer
29
+ model = AutoModelForCausalLM.from_pretrained("zenlm/zen-pro", torch_dtype="auto")
30
+ tokenizer = AutoTokenizer.from_pretrained("zenlm/zen-pro")
31
  messages = [{"role": "user", "content": "Your domain-specific prompt here"}]
32
  text = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
33
  inputs = tokenizer(text, return_tensors="pt").to(model.device)