Update README.md
Browse files
README.md
CHANGED
|
@@ -106,7 +106,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, Conversa
|
|
| 106 |
|
| 107 |
tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load tokenizer
|
| 108 |
tokenizer.padding_side = 'left'
|
| 109 |
-
model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer_full_end_to_end_s2', revision=None) # Load model
|
| 110 |
pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)
|
| 111 |
|
| 112 |
inputs = ["What volleyball exercises should I do I'm almost in high school and i do volleyball excellence five times a week (basically an advanced class in school with experienced volleyball coaches) , we have 2-3 skill training sessions a week which i feel like isn't enough for me as I would like to improve my skills almost every day.\n\n​\n\nWhat i wanted to know was what setting, digging, serving and spiking exercises could i do that would help me improve all of my skills (I have a large area to practice all these things so space isn't an issue)."]
|
|
|
|
| 106 |
|
| 107 |
tokenizer = AutoTokenizer.from_pretrained('fineinstructions/query_templatizer', revision=None) # Load tokenizer
|
| 108 |
tokenizer.padding_side = 'left'
|
| 109 |
+
model = AutoModelForCausalLM.from_pretrained('fineinstructions/query_templatizer_full_end_to_end_s2', revision=None, device_map='auto') # Load model
|
| 110 |
pipe = pipeline('text-generation', model=model, tokenizer=tokenizer, pad_token_id=tokenizer.pad_token_id, return_full_text=False)
|
| 111 |
|
| 112 |
inputs = ["What volleyball exercises should I do I'm almost in high school and i do volleyball excellence five times a week (basically an advanced class in school with experienced volleyball coaches) , we have 2-3 skill training sessions a week which i feel like isn't enough for me as I would like to improve my skills almost every day.\n\n​\n\nWhat i wanted to know was what setting, digging, serving and spiking exercises could i do that would help me improve all of my skills (I have a large area to practice all these things so space isn't an issue)."]
|