Upload anytext.py

anytext.py

@@ -469,6 +469,8 @@ class TextEmbeddingModule(nn.Module):
         super().__init__()
         # TODO: Learn if the recommended font file is free to use
         self.font = ImageFont.truetype(font_path, 60)
+        self.use_fp16 = use_fp16
+        self.device = device
         self.frozen_CLIP_embedder_t3 = FrozenCLIPEmbedderT3(device=device, use_fp16=use_fp16)
         self.embedding_manager = EmbeddingManager(self.frozen_CLIP_embedder_t3, use_fp16=use_fp16)
         rec_model_dir = "./text_embedding_module/OCR/ppv3_rec.pth"