Spaces:

arcee-ai
/

mergekit-gui

Running on A100

Set factory_reboot to True.

#36

by xi0v - opened Jan 25

←

Files changed (1) hide show

app.py CHANGED Viewed

@@ -190,7 +190,7 @@ def merge(yaml_config: str, hf_token: str, repo_name: str) -> Iterable[List[Log]
 # This is workaround. As the space always getting stuck.
 def _restart_space():
-    huggingface_hub.HfApi().restart_space(repo_id="arcee-ai/mergekit-gui", token=COMMUNITY_HF_TOKEN, factory_reboot=False)
 # Run garbage collection every hour to keep the community org clean.
 # Empty models might exists if the merge fails abruptly (e.g. if user leaves the Space).
 def _garbage_remover():

 # This is workaround. As the space always getting stuck.
 def _restart_space():
+    huggingface_hub.HfApi().restart_space(repo_id="arcee-ai/mergekit-gui", token=COMMUNITY_HF_TOKEN, factory_reboot=True)
 # Run garbage collection every hour to keep the community org clean.
 # Empty models might exists if the merge fails abruptly (e.g. if user leaves the Space).
 def _garbage_remover():