modified: app.py

app.py

@@ -75,6 +75,9 @@ class FluxEditor:
         print(f"Inital_t5_device: {self.t5.hf_module.device}")
         print(f"Inital_clip_device: {self.clip.hf_module.device}")
         print(f"Inital_flow_model: {self.model.img_in.weight.device}")
+        print(f"Inital_flow_model self.model.img_in: {self.model.img_in}")
+        print(f"Inital_flow_model self.model.time_in.out_layer.weight: {self.model.time_in.out_layer.weight}")
+
 
         # clear history
         if os.path.exists("history_gradio/history.safetensors"):