update rl model

app.py

@@ -55,8 +55,10 @@ flux_pipeline.text_encoder.to(device).to(torch.bfloat16)
 flux_pipeline.text_encoder_2.to(device).to(torch.bfloat16)
 
 # 替换 transformer 权重
-ckpt_path = hf_hub_download("NoobDoge/Multi_Ref_Model", "full_ema_model.safetensors")
-flux_pipeline.transformer = FluxTransformer2DModel.from_single_file(ckpt_path, torch_dtype=torch.bfloat16)
+ckpt_path = hf_hub_download("NoobDoge/Multi_Ref_Model", "full_model.safetensors")
+new_weight = load_file(ckpt_path)
+flux_pipeline.transformer.load_state_dict(new_weight)
+# flux_pipeline.transformer = FluxTransformer2DModel.from_single_file(ckpt_path, torch_dtype=torch.bfloat16)
 flux_pipeline.transformer.to(device).to(torch.bfloat16)
 
 # ---------------------------

gpt.zip

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:283b66020859309e45de2dfdd27da59e0eda8fa3f4d65ea6e1cf5a514b7978ce
+size 94721913