search

AmericanPresidentJimmyCarter / simple-flux-lora-training

MIT License
54 stars 5 forks source link

/root/SimpleTuner/.venv/lib/python3.11/site-packages/nvidia/nvjitlink/lib Unknown optimizer requested: adamw_bf16 #1

Closed Blackfox2012 closed 3 weeks ago

Blackfox2012 commented 3 weeks ago

/root/SimpleTuner/.venv/lib/python3.11/site-packages/nvidia/nvjitlink/lib Unknown optimizer requested: adamw_bf16

Blackfox2012 commented 3 weeks ago

when bash train.sh

Blackfox2012 commented 3 weeks ago

running pytorch/pytorch_2.4.0-cuda12.4-cudnn9-devel/jupyter

AmericanPresidentJimmyCarter commented 3 weeks ago

https://github.com/bghira/SimpleTuner/blob/1380f113e67ee80d215c31f4aefe089892936da0/train.sh#L256-L258

That error message comes from here and it looks correct. I'll ask @bghira. You can also come to the discord server: https://discord.com/invite/cSmvcU9Me9

bghira commented 3 weeks ago

try bash -x train.sh

AmericanPresidentJimmyCarter commented 3 weeks ago

Haven't heard anything back so closing.