Upload anytext.py
Browse files- anytext.py +4 -4
anytext.py
CHANGED
|
@@ -742,10 +742,10 @@ class TextEmbeddingModule(ModelMixin, ConfigMixin):
|
|
| 742 |
def __init__(self, font_path, use_fp16=False, device="cpu"):
|
| 743 |
super().__init__()
|
| 744 |
self.font = ImageFont.truetype(font_path, 60)
|
| 745 |
-
self.use_fp16 = use_fp16
|
| 746 |
-
self.device = device
|
| 747 |
-
self.frozen_CLIP_embedder_t3 = FrozenCLIPEmbedderT3(device=device, use_fp16=use_fp16)
|
| 748 |
-
self.embedding_manager = EmbeddingManager(self.frozen_CLIP_embedder_t3
|
| 749 |
self.text_predictor = create_predictor(device=device, use_fp16=use_fp16).eval()
|
| 750 |
args = {}
|
| 751 |
args["rec_image_shape"] = "3, 48, 320"
|
|
|
|
| 742 |
def __init__(self, font_path, use_fp16=False, device="cpu"):
|
| 743 |
super().__init__()
|
| 744 |
self.font = ImageFont.truetype(font_path, 60)
|
| 745 |
+
# self.use_fp16 = use_fp16
|
| 746 |
+
# self.device = device
|
| 747 |
+
self.frozen_CLIP_embedder_t3 = FrozenCLIPEmbedderT3()#device=device, use_fp16=use_fp16)
|
| 748 |
+
self.embedding_manager = EmbeddingManager(self.frozen_CLIP_embedder_t3)#, use_fp16=use_fp16)
|
| 749 |
self.text_predictor = create_predictor(device=device, use_fp16=use_fp16).eval()
|
| 750 |
args = {}
|
| 751 |
args["rec_image_shape"] = "3, 48, 320"
|