Update README.md
Browse files
README.md
CHANGED
|
@@ -52,7 +52,9 @@ output_path = "v1_final.idf"
|
|
| 52 |
|
| 53 |
# Load the EPlus-LLM model
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
|
| 55 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("EPlus-LLM/EPlus-LLMv1"
|
|
|
|
|
|
|
| 56 |
|
| 57 |
# Generation config
|
| 58 |
generation_config = model.generation_config
|
|
|
|
| 52 |
|
| 53 |
# Load the EPlus-LLM model
|
| 54 |
tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-large")
|
| 55 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("EPlus-LLM/EPlus-LLMv1"
|
| 56 |
+
# , force_download=True # If you cannot download the model
|
| 57 |
+
)
|
| 58 |
|
| 59 |
# Generation config
|
| 60 |
generation_config = model.generation_config
|