[HUB] bug is_jit.

YannDubs · Jul 7, 2021 · 6b604dd · 6b604dd
1 parent ad3fb3d
commit 6b604dd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/hub/compressor.py b/hub/compressor.py
@@ -36,7 +36,7 @@ def __init__(
         device="cuda" if torch.cuda.is_available() else "cpu",
     ):
         super().__init__()
-        model, self.preprocess = clip.load("ViT-B/32", jit=jit, device=device)
+        model, self.preprocess = clip.load("ViT-B/32", jit=is_jit, device=device)
         self.clip = model.visual
 
         self.z_dim = 512

diff --git a/hubconf.py b/hubconf.py
@@ -12,7 +12,7 @@
 
 from hub import ClipCompressor as _ClipCompressor
 
-PATH = "https://github.com/YannDubs/lossyless/releases/download/v0.1-alpha/beta{beta:0.0e}_factorized_rate.pt"
+PATH = "https://github.com/YannDubs/lossyless/releases/download/v1.0/beta{beta:0.0e}_factorized_rate.pt"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 # TODO: add JIT once https://github.com/InterDigitalInc/CompressAI/issues/72 is resolved