Python3.12 - on linux discovery issues

It seems I'm missing some modules when I run the experiements:

~/kblam/experiments$ python train.py --verbose --dataset synthetic_data --N 120000 --B 20 --total_steps 601  --encoder_spec OAI --use_oai_embd --key_embd_src key --use_data_aug

Traceback (most recent call last):
  File "/home/ubuntu/kblam/experiments/train.py", line 30, in <module>
    from kblam.models.llama3_model import KblamLlamaForCausalLM
  File "/home/ubuntu/kblam/src/kblam/models/llama3_model.py", line 43, in <module>
    from transformers.models.llama.modeling_llama import (
ImportError: cannot import name 'LlamaDynamicNTKScalingRotaryEmbedding' from 'transformers.models.llama.modeling_llama' (/home/ubuntu/kblam/lib/python3.12/site-packages/transformers/models/llama/modeling_llama.py)

• My transformers version is the 4.48.0 as per the pyproject.toml file.
• Running on LINUX in a venv
• Huggingface key is valid and connected (read only)
• Python 3.12 < Could that be the culprit?

Reproduce the command from experiments/train.py and inspect the import in src/kblam/models/llama3_model.py alongside the transformers version declared in pyproject.toml. Determine whether Python 3.12 or the installed Transformers package explains the missing symbol, then verify the expected supported environment and training command.