AjayP13 commited on
Commit
98c0aab
·
verified ·
1 Parent(s): 6402032

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -27,7 +27,7 @@ pipeline_tag: text-generation
27
  ```python3
28
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, Conversation
29
 
30
- tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer_full_end_to_end_s2', revision=None) # Load tokenizer
31
  tokenizer.padding_side = 'left'
32
  model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer_full_end_to_end_s2', revision=None) # Load model
33
  pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)
 
27
  ```python3
28
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, Conversation
29
 
30
+ tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load tokenizer
31
  tokenizer.padding_side = 'left'
32
  model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer_full_end_to_end_s2', revision=None) # Load model
33
  pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)