gitextract_vrpvpauy/

├── .gitignore
├── LICENSE
├── README.md
├── gpu_mem_track.py
├── inference.gemma.infini.py
├── infini_gemma/
│   ├── __init__.py
│   ├── configuration_infini_gemma.py
│   └── modeling_infini_gemma.py
├── infini_llama/
│   ├── __init__.py
│   └── modeling_infini_llama.py
├── modeling_gemma.py
├── original_llama.py
├── requirements.txt
├── test_basic.infini.py
├── test_basic.py
├── test_basic.trained.py
├── test_model_to_hf.py
├── test_train.small.gemma.infini.py
├── test_train.small.gemma.py
├── train.gemma.infini.noclm.1Mseq.sh
├── train.gemma.infini.noclm.32k.sh
├── train.gemma.infini.noclm.py
├── train.gemma.infini.noclm.sh
├── train.llama.infini.noclm.1Mseq.sh
└── train.llama.infini.noclm.py