Update app.py

app.py

@@ -23,6 +23,17 @@ pipe.scheduler.config.time_shift = 10
 pipe.enable_model_cpu_offload()
 pipe = pipe.to(device)
 
+
+transformer2 = FluxTransformer2DModel.from_pretrained(
+    bfl_repo, subfolder="transformer", torch_dtype=dtype
+)
+pipe2 = FluxPipeline.from_pretrained(bfl_repo, transformer=None, torch_dtype=dtype)
+pipe2.transformer = transformer2
+pipe2.scheduler.config.use_dynamic_shifting = False
+pipe2.scheduler.config.time_shift = 10
+pipe2.enable_model_cpu_offload()
+pipe2 = pipe.to(device)
+
 pipe.load_lora_weights(
     "Huage001/URAE",
     weight_name="urae_2k_adapter.safetensors",