address #17

Author: lucidrains

perfusion_pytorch/embedding.py

@@ -269,7 +269,7 @@ def forward(
         if exists(superclass_text_embeds):
             superclass_text_enc = self.text_transformer(superclass_text_embeds)
 
-        return EmbeddingReturn(text_enc, superclass_text_embeds, text_mask, concept_indices)
+        return EmbeddingReturn(text_enc, superclass_text_enc, text_mask, concept_indices)
 
 # merging multiple embedding wrappers (with one concepts) into a merged embedding wrapper with multiple concepts
 

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'perfusion-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.1.22',
+  version = '0.1.23',
   license='MIT',
   description = 'Perfusion - Pytorch',
   author = 'Phil Wang',