modified: app.py

app.py

@@ -64,7 +64,7 @@ class FluxEditor:
 
         # init all components
         self.clip = load_clip(self.device)
-        self.t5 = load_t5(self.device, max_length=256 if self.name == "flux-schnell" else 512)
+        self.t5 = load_t5(self.device, max_length=256 if self.name == "flux-schnell" else 77)
         self.model = load_flow_model(self.name, self.device)
         self.ae = load_ae(self.name, self.device)
         self.t5.eval()

src/flux/modules/conditioner.py

@@ -50,8 +50,8 @@ class HFEmbedder(nn.Module):
         assert input_ids.shape[1] == self.max_length, f"Sequence length {input_ids.shape[1]} does not match max_length {self.max_length}"
         print(input_ids)
 
-        print(f"self.tokenizer.vocab_size: {self.tokenizer.vocab_size}")  # Debug
-        print(f"self.hf_module.config.vocab_size: {self.hf_module.config.vocab_size}")  # Debug
+        print(f"{flag: }self.tokenizer.vocab_size: {self.tokenizer.vocab_size}")  # Debug
+        print(f"{flag: }self.hf_module.config.vocab_size: {self.hf_module.config.vocab_size}")  # Debug
 
         outputs = self.hf_module(
             input_ids=input_ids.to(self.hf_module.device),