requirements.txt

transformers>=4.38.1  # Specify a minimum version that supports Gemma and Flash Attention 2
gradio>=4.0.0        # Use a recent version of Gradio for gr.ChatInterface
torch>=2.1.0         # Use a recent version of PyTorch (2.1+ is recommended)
accelerate>=0.25.0   # For efficient model loading and device management
safetensors          # For loading .safetensors model files