Update llama.py

This commit is contained in:
Daniel Han 2025-03-06 03:46:00 -08:00
commit 2ea8bb041e

View file

@ -1897,8 +1897,6 @@ class FastLlamaModel:
"_inner_training_loop",
"_fast_inner_training_loop", 1,
)
Trainer._inner_training_loop = _fast_inner_training_loop
inner_training_loop = inner_training_loop.replace(
"is_torch_tpu_available()",
"False",