make style

Author: tolgacangoz

examples/research_projects/anytext/auxiliary_latent_module.py

@@ -29,7 +29,14 @@ def retrieve_latents(
 
 class AuxiliaryLatentModule(nn.Module):
     def __init__(
-        self, font_path, glyph_channels=1, position_channels=1, model_channels=320, vae=None, device="cpu", use_fp16=False
+        self,
+        font_path,
+        glyph_channels=1,
+        position_channels=1,
+        model_channels=320,
+        vae=None,
+        device="cpu",
+        use_fp16=False,
     ):
         super().__init__()
         self.font = ImageFont.truetype(font_path, 60)
@@ -78,12 +85,8 @@ def __init__(
 
         self.fuse_block = nn.Conv2d(256 + 64 + 4, model_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.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:

examples/research_projects/anytext/ocr_recog/RecModel.py

@@ -35,6 +35,7 @@ def load_3rd_state_dict(self, _3rd_name, _state):
 
     def forward(self, x):
         import torch
+
         x = x.to(torch.float32)
         x = self.backbone(x)
         x = self.neck(x)

examples/research_projects/anytext/recognizer.py

@@ -13,7 +13,6 @@
 import torch.nn.functional as F
 from easydict import EasyDict as edict
 from ocr_recog.RecModel import RecModel
-from safetensors.torch import load_file
 from skimage.transform._geometric import _umeyama as get_sym_mat