iamleonie commited on
Commit
7453bca
·
verified ·
1 Parent(s): f6a5d17

Update README.md

Browse files

remove chat usage example incl. sampling params from base model

Files changed (1) hide show
  1. README.md +0 -20
README.md CHANGED
@@ -88,26 +88,6 @@ model = AutoModelForCausalLM.from_pretrained(
88
  # attn_implementation="flash_attention_2" <- uncomment on compatible GPU
89
  )
90
  tokenizer = AutoTokenizer.from_pretrained(model_id)
91
- streamer = TextStreamer(tokenizer, skip_prompt=True, skip_special_tokens=True)
92
-
93
- prompt = "What is C. elegans?"
94
-
95
- input_ids = tokenizer.apply_chat_template(
96
- [{"role": "user", "content": prompt}],
97
- add_generation_prompt=True,
98
- return_tensors="pt",
99
- tokenize=True,
100
- ).to(model.device)
101
-
102
- output = model.generate(
103
- input_ids,
104
- do_sample=True,
105
- temperature=0.3,
106
- min_p=0.15,
107
- repetition_penalty=1.05,
108
- max_new_tokens=512,
109
- streamer=streamer,
110
- )
111
  ```
112
 
113
  ## 🔧 Fine-Tuning
 
88
  # attn_implementation="flash_attention_2" <- uncomment on compatible GPU
89
  )
90
  tokenizer = AutoTokenizer.from_pretrained(model_id)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
91
  ```
92
 
93
  ## 🔧 Fine-Tuning