tolgacangoz
/

anytext

Model card Files Files and versions Community

tolgacangoz commited on 5 days ago

Commit

be35f0f

·

verified ·

1 Parent(s): 0236f91

Upload anytext.py

Files changed (1) hide show

anytext.py +1 -1

anytext.py CHANGED Viewed

@@ -1029,7 +1029,7 @@ class TextEmbeddingModule(ModelMixin, ConfigMixin):
         self.frozen_CLIP_embedder_t3 = self.frozen_CLIP_embedder_t3.to(*args, **kwargs)
         self.embedding_manager = self.embedding_manager.to(*args, **kwargs)
         self.text_predictor = self.text_predictor.to(*args, **kwargs)
-        self.device = self.frozen_CLIP_embedder_t3.device
         return self

         self.frozen_CLIP_embedder_t3 = self.frozen_CLIP_embedder_t3.to(*args, **kwargs)
         self.embedding_manager = self.embedding_manager.to(*args, **kwargs)
         self.text_predictor = self.text_predictor.to(*args, **kwargs)
+        # self.device = self.frozen_CLIP_embedder_t3.device
         return self