Specify ref for forked torch.hub repos.

2026-04-03 03:00:17 -04:00 · 2023-10-19 10:49:28 -05:00
parent 0b3c4c029d
commit 791553762b
1 changed files with 2 additions and 2 deletions
--- a/apps/stable_diffusion/src/utils/stencils/zoe/init.py
+++ b/apps/stable_diffusion/src/utils/stencils/zoe/init.py
@@ -25,13 +25,13 @@ class ZoeDetector:
                    f.write(chunk)

        midas = torch.hub.load(
-            "gpetters94/MiDaS",
+            "gpetters94/MiDaS:master",
            "DPT_BEiT_L_384",
            pretrained=False,
            force_reload=False,
        )
        model = torch.hub.load(
-            "AyaanShah2204/ZoeDepth",
+            "monorimet/ZoeDepth:torch_update",
            "ZoeD_N",
            pretrained=False,
            force_reload=False,