Update README.md

README.md

@@ -58,7 +58,7 @@ def predict_NuExtract(model,tokenizer,text, schema,example = ["","",""]):
     return output.split("<|output|>")[1].split("<|end-output|>")[0]
 
 
-model = AutoModelForCausalLM.from_pretrained("numind/NuExtract", trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained("numind/NuExtract", trust_remote_code=True,torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained("numind/NuExtract", trust_remote_code=True)
 
 #model.to("cuda")