Error Sample Code
#2
by
joon09
- opened
Thank you for sharing the good materials.
However, the sample code is NuExtract, so an error appears.
I would appreciate it if you could provide me with the sample code again.
Thanks for the feedback.
Fixed it, need to load the model with bf16 (AutoModelForCausalLM.from_pretrained("numind/NuExtract", trust_remote_code=True,torch_dtype=torch.bfloat16))
Alexandre-Numind
changed discussion status to
closed