natolambert
/

unet-rl-hopper-tmp

Inference Endpoints

Model card Files Files and versions Community

Nathan Lambert commited on Jun 21, 2022

Commit

cfb50db

•

1 Parent(s): 947698a

update config

Files changed (1) hide show

config.json +5 -1

config.json CHANGED Viewed

@@ -1,5 +1,9 @@
 {
   "_class_name": "TemporalUnet",
   "_diffusers_version": "0.0.4",
-  "name_or_path": "/Users/nato/Downloads/tmp_models/"
 }

 {
   "_class_name": "TemporalUnet",
   "_diffusers_version": "0.0.4",
+  "horizon": 128,
+  "dim": 32,
+  "dim_mults": [1, 4, 8],
+  "transition_dim": 14,
+  "cond_dim": 3
 }