Update Kinetix/kinetix/editor.py

Kinetix/kinetix/editor.py

@@ -1,7 +1,7 @@
 import itertools
 import time
 from timeit import default_timer as tmr
-
+import asyncio
 import optax
 from PIL import Image
 from flax.serialization import to_state_dict
@@ -2642,7 +2642,7 @@ class Editor:
 
 
 @hydra.main(version_base=None, config_path="../configs", config_name="editor")
-def main(config):
+async def main(config):
     config = normalise_config(OmegaConf.to_container(config), "EDITOR", editor_config=True)
     env_params, static_env_params = generate_params_from_config(config)
     static_env_params = static_env_params.replace(frame_skip=config["frame_skip"], downscale=config["downscale"])
@@ -2667,5 +2667,6 @@ def main(config):
         clock.tick(config["fps"])
 
 
-if __name__ == "__main__":
-    main()
+'''if __name__ == "__main__":
+    main()'''
+asyncio.run(main())