Upload RoboTransferPipeline

depth_guider_net/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "GuiderNet",
-  "_diffusers_version": "0.33.1",
-  "_name_or_path": "test_pipe_weights/depth_guider_net",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/shared_disk/users/wenkang.qin/checkpoints/robotransfer-low-resolution/depth_guider_net",
   "in_channels": 1,
   "mid_channels": 4,
   "out_channels": 4

image_encoder/config.json

@@ -18,5 +18,5 @@
   "patch_size": 14,
   "projection_dim": 1024,
   "torch_dtype": "float32",
-  "transformers_version": "4.52.4"
+  "transformers_version": "4.53.1"
 }

model_index.json

@@ -1,9 +1,9 @@
 {
   "_class_name": "RoboTransferPipeline",
-  "_diffusers_version": "0.33.1",
-  "_name_or_path": "test_pipe_weights",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/shared_disk/users/wenkang.qin/checkpoints/robotransfer-low-resolution",
   "depth_guider_net": [
-    "models.guider",
+    "robotransfer.models.guider",
     "GuiderNet"
   ],
   "feature_extractor": [
@@ -15,7 +15,7 @@
     "CLIPVisionModelWithProjection"
   ],
   "normal_guider_net": [
-    "models.guider",
+    "robotransfer.models.guider",
     "GuiderNet"
   ],
   "scheduler": [

normal_guider_net/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "GuiderNet",
-  "_diffusers_version": "0.33.1",
-  "_name_or_path": "test_pipe_weights/normal_guider_net",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/shared_disk/users/wenkang.qin/checkpoints/robotransfer-low-resolution/normal_guider_net",
   "in_channels": 3,
   "mid_channels": 4,
   "out_channels": 4

scheduler/scheduler_config.json

@@ -1,6 +1,6 @@
 {
   "_class_name": "EulerDiscreteScheduler",
-  "_diffusers_version": "0.33.1",
+  "_diffusers_version": "0.34.0",
   "beta_end": 0.012,
   "beta_schedule": "scaled_linear",
   "beta_start": 0.00085,

unet/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "UNetSpatioTemporalConditionModel",
-  "_diffusers_version": "0.33.1",
-  "_name_or_path": "test_pipe_weights/unet",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/shared_disk/users/wenkang.qin/checkpoints/robotransfer-low-resolution/unet",
   "addition_time_embed_dim": 256,
   "block_out_channels": [
     320,

vae/config.json

@@ -1,7 +1,7 @@
 {
   "_class_name": "AutoencoderKLTemporalDecoder",
-  "_diffusers_version": "0.33.1",
-  "_name_or_path": "test_pipe_weights/vae",
+  "_diffusers_version": "0.34.0",
+  "_name_or_path": "/shared_disk/users/wenkang.qin/checkpoints/robotransfer-low-resolution/vae",
   "block_out_channels": [
     128,
     256,