DeathReaper0965 commited on
Commit
78ec975
·
verified ·
1 Parent(s): b91455d

add TextStreamer

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ import torch
27
 
28
  from peft import PeftModel, PeftConfig
29
 
30
- from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
31
 
32
 
33
  config = PeftConfig.from_pretrained("DeathReaper0965/Qwen2.5-3B-Inst-SQL-Reasoning-GRPO")
 
27
 
28
  from peft import PeftModel, PeftConfig
29
 
30
+ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, TextStreamer
31
 
32
 
33
  config = PeftConfig.from_pretrained("DeathReaper0965/Qwen2.5-3B-Inst-SQL-Reasoning-GRPO")