Alexandre-Numind commited on
Commit
0bf8048
1 Parent(s): 09be9ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -59,7 +59,7 @@ def predict_NuExtract(model,tokenizer,text, schema,example = ["","",""]):
59
  model = AutoModelForCausalLM.from_pretrained("numind/NuExtract", trust_remote_code=True)
60
  tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract", trust_remote_code=True)
61
 
62
- #model.to("cuda")
63
 
64
  model.eval()
65
 
 
59
  model = AutoModelForCausalLM.from_pretrained("numind/NuExtract", trust_remote_code=True)
60
  tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract", trust_remote_code=True)
61
 
62
+ model.to("cuda")
63
 
64
  model.eval()
65