rewardfm_eval_ui / requirements.txt
jesbu1's picture
gradio installation is moved and a fallback is added
181f3ff
raw
history blame contribute delete
824 Bytes
# Requirements for RFM Eval UI Gradio App
# Core dependencies
matplotlib>=3.5.0
numpy>=1.21.0
torch>=2.0.0
PyYAML>=6.0
Pillow>=9.0.0
# HuggingFace
huggingface-hub>=0.16.0
transformers>=4.30.0
# Sentence transformers (for ReWiND models)
sentence-transformers>=2.2.0
# Qwen VL utilities
qwen-vl-utils
# Video processing
opencv-python-headless>=4.5.0
decord>=0.6.0 # For video frame extraction (same as preprocess_datasets.py)
# Development tools (optional, for auto-reload)
watchfiles # For file watching during development
# RFM package (installed from git repository)
# For local development, you can also install with: pip install -e ../ (from parent directory)
git+https://github.com/aliang8/reward_fm.git@93b1ad4b5a530fb32c234bf926b659105e676d00
# Make sure a newer version of gradio is installed
gradio==4.44.0