more debug

app.py

@@ -9,6 +9,13 @@ from pathlib import Path
 from serve.constants import SERVER_PORT, ROOT_PATH, ELO_RESULTS_DIR
 from model.pre_download import pre_download_all_models, pre_download_video_models
 
+def debug_packages():
+    import pkg_resources
+
+    installed_packages = pkg_resources.working_set
+    for package in installed_packages:
+        print(f"{package.key}=={package.version}")
+
 def build_combine_demo(models, elo_results_file, leaderboard_table_file):
 
     with gr.Blocks(
@@ -99,6 +106,8 @@ if __name__ == "__main__":
     elo_results_dir = ELO_RESULTS_DIR
     models = ModelManager()
 
+    debug_packages()
+
     pre_download_all_models()
     
     elo_results_file, leaderboard_table_file = load_elo_results(elo_results_dir)

requirements.txt

@@ -27,7 +27,6 @@ torch-fidelity>=0.3.0
 setuptools>=59.5.0
 transformers
 torchmetrics>=0.6.0
-lpips
 kornia>=0.6
 diffusers>=0.18.0
 accelerate>=0.20.3
@@ -66,7 +65,7 @@ pre-commit
 pyav
 pyarrow
 tensorboard
-timm~=0.9
+timm
 wandb
 pandarallel
 kaleido