ruslanmv commited on
Commit
d34ac5f
1 Parent(s): ec1b39e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -46,7 +46,7 @@ tokenizer = AutoTokenizer.from_pretrained(finetuned_model, trust_remote_code=Tru
46
  # Set pad_token_id to eos_token_id
47
  model_pretrained.config.pad_token_id = tokenizer.eos_token_id
48
 
49
- pipe = pipeline(task="text-generation", model=model_pretrained, tokenizer=tokenizer, max_length=500)
50
 
51
  def build_prompt(question):
52
  prompt = f"[INST]@Enlighten. [/INST] {question}"
 
46
  # Set pad_token_id to eos_token_id
47
  model_pretrained.config.pad_token_id = tokenizer.eos_token_id
48
 
49
+ pipe = pipeline(task="text-generation", model=model_pretrained, tokenizer=tokenizer, max_length=100)
50
 
51
  def build_prompt(question):
52
  prompt = f"[INST]@Enlighten. [/INST] {question}"