add video UI

2026-03-19 06:32:27 +00:00 · 2024-01-09 19:07:16 +08:00
parent 49db42bf1f
commit cd6fa30c9a
9 changed files with 294 additions and 20 deletions
--- a/diffsynth/models/init.py
+++ b/diffsynth/models/init.py
@@ -189,6 +189,7 @@ class ModelManager:
                    model.to(device)
            else:
                self.model[component].to(device)
+        torch.cuda.empty_cache()

    def get_model_with_model_path(self, model_path):
        for component in self.model_path: