Text-to-Image
Diffusers
Safetensors
tolgacangoz commited on
Commit
17cf7e7
·
verified ·
1 Parent(s): c747cba

Upload anytext.py

Browse files
Files changed (1) hide show
  1. anytext.py +1 -1
anytext.py CHANGED
@@ -476,7 +476,7 @@ class TextEmbeddingModule(nn.Module):
476
  args["rec_image_shape"] = "3, 48, 320"
477
  args["rec_batch_num"] = 6
478
  args["rec_char_dict_path"] = "./text_embedding_module/OCR/ppocr_keys_v1.txt"
479
- args["use_fp16"] = self.use_fp16
480
  self.embedding_manager.recog = TextRecognizer(args, self.text_predictor)
481
 
482
  @torch.no_grad()
 
476
  args["rec_image_shape"] = "3, 48, 320"
477
  args["rec_batch_num"] = 6
478
  args["rec_char_dict_path"] = "./text_embedding_module/OCR/ppocr_keys_v1.txt"
479
+ args["use_fp16"] = use_fp16
480
  self.embedding_manager.recog = TextRecognizer(args, self.text_predictor)
481
 
482
  @torch.no_grad()