example crushs
#110
by
brand17
- opened
I am trying this example on TPU:
https://huggingface.co/google/gemma-7b/blob/main/examples/example_fsdp.py
Please check my notebook here: https://colab.research.google.com/drive/1bEySvl3HagayCXQPxHE53Z-fM5UyKuaV?usp=sharing
I see the folloowing error in the log:ERROR: Unknown command line flag 'xla_latency_hiding_scheduler_rerun'
Hi @brand17 ,
We ran the provided notebook but were unable to reproduce the issue or error. The code executed successfully. We used Accelerator = TPU VM v3-8, and the library versions were as follows:
torch = 2.3.0+cu121
torch_xla = 2.3.0+libtpu
peft = 0.12.0
trl = 0.9.6
If the issue still persists, please let us know
Thank you.