# Requirements for ComfyUI-OminiKontext
# These are in addition to ComfyUI's base requirements

# Core dependencies
torch>=2.0.0
diffusers @ git+https://github.com/huggingface/diffusers.git
transformers>=4.35.0
accelerate>=0.25.0

# Additional requirements
peft>=0.7.0
safetensors>=0.4.0
einops>=0.7.0
protobuf>=4.0.0
sentencepiece>=0.1.99

# Optional for training features
lightning>=2.0.0
prodigyopt>=1.0