Hugging Face's logo

Spaces:
ksort
/
K-Sort-Arena
Running on Zero

App Files Community
2
ksort commited on
Commit
6d73014
·
1 Parent(s): d00d439

Fix bug

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -113,7 +113,7 @@ class ModelManager:
113
 
114
  root_dir = SSH_CACHE_OPENSOURCE
115
  for name in model_names:
116
- if "Runway-Gen3" in name or "Runway-Gen2" in name or "Pika-v1.0" in name:
117
  root_dir = SSH_CACHE_ADVANCE
118
  elif "Pika-beta" in name:
119
  root_dir = SSH_CACHE_PIKA
 
113
 
114
  root_dir = SSH_CACHE_OPENSOURCE
115
  for name in model_names:
116
+ if "Runway-Gen3" in name or "Runway-Gen2" in name or "Pika-v1.0" in name or "KLing-v1.0" in name:
117
  root_dir = SSH_CACHE_ADVANCE
118
  elif "Pika-beta" in name:
119
  root_dir = SSH_CACHE_PIKA