Update run.py

2024-09-25 23:28:07 +03:00 · 2024-09-20 17:49:56 +08:00
parent 5641ce97c3
commit b07fbdd6df
1 changed files with 0 additions and 1 deletions
--- a/run.py
+++ b/run.py
@@ -19,7 +19,6 @@ class DepthCrafterDemo:
    ):
        unet = DiffusersUNetSpatioTemporalConditionModelDepthCrafter.from_pretrained(
            unet_path,
-            subfolder="unet",
            low_cpu_mem_usage=True,
            torch_dtype=torch.float16,
        )