Update README.md
Browse files
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/
|
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)
|