setup_dependencies.py

import subprocess

def install_dependencies():
    dependencies = [
        "packaging",
        "setuptools",
        "transformers==4.32.0",
        "safetensors",
        "tiktoken",
        "flask",
        "torch",
        "gunicorn==20.1.0",
        "flash-attn"
    ]
    
    for package in dependencies:
        subprocess.run([sys.executable, "-m", "pip", "install", "--no-cache-dir", "--upgrade", package])

install_dependencies()