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'

Google org

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.

Sign up or log in to comment