0.0.9

lucidrains · Sep 2, 2023 · 8bb2f47 · 8bb2f47
1 parent 6ce3826
commit 8bb2f47
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/coca_pytorch/coca_pytorch.py b/coca_pytorch/coca_pytorch.py
@@ -110,8 +110,8 @@ def __init__(self, dim, dim_head=64, heads=8, ff_mult=4):
 
         # for caching causal mask and rotary embeddings
 
-        self.register_buffer("mask", None, persistent=False)
-        self.register_buffer("pos_emb", None, persistent=False)
+        self.mask = None
+        self.pos_emb = None
 
     def get_mask(self, n, device):
         if self.mask is not None and self.mask.shape[-1] >= n:

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'CoCa-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.0.8',
+  version = '0.0.9',
   license='MIT',
   description = 'CoCa, Contrastive Captioners are Image-Text Foundation Models - Pytorch',
   author = 'Phil Wang',