change checkpoint

text_encoder/config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "ddpo-aesthetic/text_encoder",
+  "_name_or_path": "flax_tmp/text_encoder",
   "architectures": [
     "CLIPTextModel"
   ],

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNet2DConditionModel",
   "_diffusers_version": "0.16.1",
-  "_name_or_path": "ddpo-aesthetic/unet",
+  "_name_or_path": "flax_tmp/unet",
   "act_fn": "silu",
   "addition_embed_type": null,
   "addition_embed_type_num_heads": 64,

unet/diffusion_pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:86d0144fa1a0d535ebc458d492e8ab69653db4734b2d3c6880f2a1c6e067b49f
+oid sha256:9f3fceda0246a3606f8fc8a9df5cb9bee43aa97130c3e98c9a7252838011493a
 size 3438366373

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKL",
   "_diffusers_version": "0.16.1",
-  "_name_or_path": "ddpo-aesthetic/vae",
+  "_name_or_path": "flax_tmp/vae",
   "act_fn": "silu",
   "block_out_channels": [
     128,