numind
/

NuExtract-1.5-tiny

Text Generation

Model card Files Files and versions Community

liamcripwell commited on 2 days ago

Commit

765d9d8

•

1 Parent(s): f45af47

Update README.md

Files changed (1) hide show

README.md +1 -0

README.md CHANGED Viewed

@@ -42,6 +42,7 @@ To use the model:
 ```python
 import json
 from transformers import AutoModelForCausalLM, AutoTokenizer
 def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):

 ```python
 import json
+import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 def predict_NuExtract(model, tokenizer, texts, template, batch_size=1, max_length=10_000, max_new_tokens=4_000):