update model2

app.py

@@ -52,8 +52,8 @@ flux_pipeline.text_encoder.to(device).to(torch.bfloat16)
 flux_pipeline.text_encoder_2.to(device).to(torch.bfloat16)
 flux_pipeline.scheduler.config.stochastic_sampling = False
 ckpt_path = hf_hub_download("NoobDoge/Multi_Ref_Model", "full_ema_model.safetensors")
-new_weight = load_file(ckpt_path)
-flux_pipeline.transformer.load_state_dict(new_weight)
+# new_weight = load_file(ckpt_path)
+flux_pipeline.transformer.from_single_file(ckpt_path, torch_dtype=torch.bfloat16)
 flux_pipeline.transformer.to(device).to(torch.bfloat16)
 
 MAX_SEED = np.iinfo(np.int32).max