tolgacangoz
/

anytext

Model card Files Files and versions

tolgacangoz commited on Feb 24

Commit

796e596

·

verified ·

1 Parent(s): caeba51

Upload anytext.py

Files changed (1) hide show

anytext.py +2 -2

anytext.py CHANGED Viewed

@@ -228,8 +228,8 @@ class EmbeddingManager(ModelMixin, ConfigMixin):
             self.proj = self.proj.to(dtype=torch.float16)
         # self.register_parameter("proj", proj)
-        placeholder_token = get_token_for_string(placeholder_string)
-        self.register_config(placeholder_token=placeholder_token)
     @torch.no_grad()
     def encode_text(self, text_info):

             self.proj = self.proj.to(dtype=torch.float16)
         # self.register_parameter("proj", proj)
+        self.placeholder_token = get_token_for_string(placeholder_string)
+        # self.register_config(placeholder_token=placeholder_token)
     @torch.no_grad()
     def encode_text(self, text_info):