modified: src/flux/modules/conditioner.py

app.py

@@ -550,7 +550,7 @@ if __name__ == "__main__":
     import argparse
     parser = argparse.ArgumentParser(description="Flux")
     parser.add_argument("--name", type=str, default="flux-dev", choices=list(configs.keys()), help="Model name")
-    parser.add_argument("--device", type=str, default="cuda" if torch.cuda.is_available() else "cpu", help="Device to use")
+    parser.add_argument("--device", type=str, default="cuda", help="Device to use")
     parser.add_argument("--offload", default=False, help="Offload model to CPU when not in use")
     parser.add_argument("--share", action="store_true", help="Create a public link to your demo")
     parser.add_argument("--port", type=int, default=9090)

src/flux/modules/conditioner.py

@@ -67,7 +67,7 @@ class HFEmbedder(nn.Module):
             padding="max_length",
             return_tensors="pt",
         )
-
+        print(f'Batch Encoding {batch_encoding}')
         encoder_type = 'clip' if self.is_clip else 't5'
         print(f'Forward pass for {encoder_type}')
         input_ids = batch_encoding["input_ids"]