entity2260
/

algebra_linear_1d

text2text-generation

text-generation-inference

Model card Files Files and versions

entity2260 commited on Jul 9, 2024

Commit

d0bd871

·

verified ·

1 Parent(s): c272d98

Update README.md

Files changed (1) hide show

README.md +3 -3

README.md CHANGED Viewed

@@ -14,10 +14,10 @@ To load the model:
 (necessary packages: !pip install transformers sentencepiece)
 ```python
 # Load model directly
-from transformers import AutoTokenizer, AutoModelForCausalLM
-tokenizer = AutoTokenizer.from_pretrained("entity2260/pythia-70m")
-model = AutoModelForCausalLM.from_pretrained("entity2260/pythia-70m")
 ```
 You can then use this model to solve algebra 1d equations into numbers.

 (necessary packages: !pip install transformers sentencepiece)
 ```python
 # Load model directly
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+tokenizer = AutoTokenizer.from_pretrained("entity2260/algebra_linear_1d")
+model = AutoModelForSeq2SeqLM.from_pretrained("entity2260/algebra_linear_1d")
 ```
 You can then use this model to solve algebra 1d equations into numbers.