Hugging Face's logo

leekwoon
/
walker2d-medium-v2-H32-T20

Diffusers
ValueGuidedDiffuserPipeline
Model card Files
xet
 Community
leekwoon commited on
Commit
ea2d3e1
·
1 Parent(s): ad4ffff

convert_diffuser_to_diffusers

Browse files
Files changed (6) hide show
  1. model_index.json +16 -0
  2. scheduler/scheduler_config.json +12 -0
  3. unet/config.json +36 -0
  4. unet/diffusion_pytorch_model.bin +3 -0
  5. value_function/config.json +32 -0
  6. value_function/diffusion_pytorch_model.bin +3 -0
model_index.json ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ValueGuidedDiffuserPipeline",
3
+ "_diffusers_version": "0.12.1",
4
+ "scheduler": [
5
+ "diffusers",
6
+ "DDPMScheduler"
7
+ ],
8
+ "unet": [
9
+ "diffusers",
10
+ "UNet1DModel"
11
+ ],
12
+ "value_function": [
13
+ "diffusers",
14
+ "UNet1DModel"
15
+ ]
16
+ }
scheduler/scheduler_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "DDPMScheduler",
3
+ "_diffusers_version": "0.12.1",
4
+ "beta_end": 0.02,
5
+ "beta_schedule": "squaredcos_cap_v2",
6
+ "beta_start": 0.0001,
7
+ "clip_sample": false,
8
+ "num_train_timesteps": 20,
9
+ "prediction_type": "sample",
10
+ "trained_betas": null,
11
+ "variance_type": "fixed_small"
12
+ }
unet/config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet1DModel",
3
+ "_diffusers_version": "0.12.1",
4
+ "act_fn": "mish",
5
+ "block_out_channels": [
6
+ 32,
7
+ 64,
8
+ 128,
9
+ 256
10
+ ],
11
+ "down_block_types": [
12
+ "DownResnetBlock1D",
13
+ "DownResnetBlock1D",
14
+ "DownResnetBlock1D",
15
+ "DownResnetBlock1D"
16
+ ],
17
+ "downsample_each_block": false,
18
+ "extra_in_channels": 0,
19
+ "flip_sin_to_cos": false,
20
+ "freq_shift": 1,
21
+ "in_channels": 23,
22
+ "layers_per_block": 1,
23
+ "mid_block_type": "MidResTemporalBlock1D",
24
+ "norm_num_groups": 8,
25
+ "out_block_type": "OutConv1DBlock",
26
+ "out_channels": 23,
27
+ "sample_rate": null,
28
+ "sample_size": 65536,
29
+ "time_embedding_type": "positional",
30
+ "up_block_types": [
31
+ "UpResnetBlock1D",
32
+ "UpResnetBlock1D",
33
+ "UpResnetBlock1D"
34
+ ],
35
+ "use_timestep_embedding": true
36
+ }
unet/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1022dfa7069c67a052d76134096c25ffe60508a4b84f7ec421a10512b3ab5a1c
3
+ size 15907449
value_function/config.json ADDED
@@ -0,0 +1,32 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "UNet1DModel",
3
+ "_diffusers_version": "0.12.1",
4
+ "act_fn": "mish",
5
+ "block_out_channels": [
6
+ 32,
7
+ 64,
8
+ 128,
9
+ 256
10
+ ],
11
+ "down_block_types": [
12
+ "DownResnetBlock1D",
13
+ "DownResnetBlock1D",
14
+ "DownResnetBlock1D",
15
+ "DownResnetBlock1D"
16
+ ],
17
+ "downsample_each_block": true,
18
+ "extra_in_channels": 0,
19
+ "flip_sin_to_cos": false,
20
+ "freq_shift": 1,
21
+ "in_channels": 23,
22
+ "layers_per_block": 1,
23
+ "mid_block_type": "ValueFunctionMidBlock1D",
24
+ "norm_num_groups": 8,
25
+ "out_block_type": "ValueFunction",
26
+ "out_channels": 23,
27
+ "sample_rate": null,
28
+ "sample_size": 65536,
29
+ "time_embedding_type": "positional",
30
+ "up_block_types": [],
31
+ "use_timestep_embedding": true
32
+ }
value_function/diffusion_pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a458a825bf222154ce5bd7585842203beaefe94224096d433ad9eab2cf0873a
3
+ size 9235257