Upload anytext.py

anytext.py

@@ -211,7 +211,7 @@ class EmbeddingManager(nn.Module):
         use_fp16=False,
     ):
         super().__init__()
-        get_token_for_string = partial(get_clip_token_for_string, self.clip_tokenizer)
+        get_token_for_string = partial(get_clip_token_for_string, embedder.clip_tokenizer)
         token_dim = 768
         self.get_recog_emb = None
         self.token_dim = token_dim
@@ -222,7 +222,7 @@ class EmbeddingManager(nn.Module):
             filename="text_embedding_module/proj.safetensors",
             cache_dir=HF_MODULES_CACHE,
         )
-        self.proj.load_state_dict(load_file(proj_dir, device=str(self.device)))
+        self.proj.load_state_dict(load_file(proj_dir, device=str(embedder.device)))
         if use_fp16:
             self.proj = self.proj.to(dtype=torch.float16)