search

SqueezeAILab / LLM2LLM

[ACL 2024] LLM2LLM: Boosting LLMs with Novel Iterative Data Enhancement
https://arxiv.org/abs/2403.15042
MIT License
155 stars 10 forks source link

expected dtype float for `end` but got dtype c10::BFloat16 #5

Open tcxia opened 7 months ago

tcxia commented 7 months ago

Running the gitcode encountered the above error, the full error is as follows: image

transformers == 4.32.1 torch==2.2.0

Vashistht commented 2 months ago

getting the same error with the newest torch and transformers==4.36.2