Text-to-Image
Diffusers
Safetensors
tolgacangoz commited on
Commit
8217bf1
·
verified ·
1 Parent(s): fa6102a

Upload anytext.py

Browse files
Files changed (1) hide show
  1. 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, 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"
 
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"