[UPDATE] Remove commented-out code and unnecessary docstring in anytext.py and anytext_controlnet.py for improved clarity

Author: tolgacangoz

examples/research_projects/anytext/anytext.py

@@ -25,6 +25,7 @@
 import os
 import re
 import sys
+import unicodedata
 from functools import partial
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
@@ -324,12 +325,6 @@ def adjust_image(box, img):
     return result
 
 
-"""
-mask: numpy.ndarray, mask of textual, HWC
-src_img: torch.Tensor, source image, CHW
-"""
-
-
 def crop_image(src_img, mask):
     box = min_bounding_rect(mask)
     result = adjust_image(box, src_img)
@@ -526,10 +521,8 @@ def get_ctcloss(self, preds, gt_text, weight):
 
 
 class TextEmbeddingModule(nn.Module):
-    # @register_to_config
     def __init__(self, font_path, use_fp16=False, device="cpu"):
         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

examples/research_projects/anytext/anytext_controlnet.py

@@ -85,15 +85,6 @@ def __init__(
 
         self.fuse_block = nn.Conv2d(256 + 64 + 4, conditioning_embedding_channels, 3, padding=1)
 
-        # self.glyph_block.load_state_dict(load_file("glyph_block.safetensors", device=str(self.device)))
-        # self.position_block.load_state_dict(load_file("position_block.safetensors", device=str(self.device)))
-        # self.fuse_block.load_state_dict(load_file("fuse_block.safetensors", device=str(self.device)))
-
-        # if use_fp16:
-        #     self.glyph_block = self.glyph_block.to(dtype=torch.float16)
-        #     self.position_block = self.position_block.to(dtype=torch.float16)
-        #     self.fuse_block = self.fuse_block.to(dtype=torch.float16)
-
     def forward(self, glyphs, positions, text_info):
         glyph_embedding = self.glyph_block(glyphs.to(self.glyph_block[0].weight.device))
         position_embedding = self.position_block(positions.to(self.position_block[0].weight.device))