Hugging Face's logo

py-img-gen
/
ncsn-mnist

Diffusers
Safetensors
Model card Files Community
shunk031 commited on
Commit
bf85952
•
1 Parent(s): 7329ad8

Upload NCSNPipeline

Browse files
Files changed (4) hide show
  1. model_index.json +4 -5
  2. scheduler/scheduler_config.json +1 -2
  3. unet/config.json +0 -1
  4. unet/diffusion_pytorch_model.safetensors +1 -1
model_index.json CHANGED
@@ -1,13 +1,12 @@
1
  {
2
- "_class_name": ["pipeline_ncsn", "NCSNPipeline"],
3
- "_name_or_path": "py-img-gen/ncsn-mnist",
4
  "_diffusers_version": "0.32.0",
5
  "scheduler": [
6
- "scheduling_ncsn",
7
- "AnnealedLangevinDynamicScheduler"
8
  ],
9
  "unet": [
10
- "unet_2d_ncsn",
11
  "UNet2DModelForNCSN"
12
  ]
13
  }
 
1
  {
2
+ "_class_name": "NCSNPipeline",
 
3
  "_diffusers_version": "0.32.0",
4
  "scheduler": [
5
+ "ncsn.scheduler.scheduling_ncsn",
6
+ "AnnealedLangevinDynamicsScheduler"
7
  ],
8
  "unet": [
9
+ "ncsn.unet.unet_2d_ncsn",
10
  "UNet2DModelForNCSN"
11
  ]
12
  }
scheduler/scheduler_config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
- "_class_name": "AnnealedLangevinDynamicScheduler",
3
  "_diffusers_version": "0.32.0",
4
- "_name_or_path": "py-img-gen/ncsn-mnist",
5
  "num_annealed_steps": 100,
6
  "num_train_timesteps": 10,
7
  "sampling_eps": 1e-05,
 
1
  {
2
+ "_class_name": "AnnealedLangevinDynamicsScheduler",
3
  "_diffusers_version": "0.32.0",
 
4
  "num_annealed_steps": 100,
5
  "num_train_timesteps": 10,
6
  "sampling_eps": 1e-05,
unet/config.json CHANGED
@@ -1,7 +1,6 @@
1
  {
2
  "_class_name": "UNet2DModelForNCSN",
3
  "_diffusers_version": "0.32.0",
4
- "_name_or_path": "py-img-gen/ncsn-mnist",
5
  "act_fn": "silu",
6
  "add_attention": true,
7
  "attention_head_dim": 8,
 
1
  {
2
  "_class_name": "UNet2DModelForNCSN",
3
  "_diffusers_version": "0.32.0",
 
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 8,
unet/diffusion_pytorch_model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4712812ff34ef7ebbbdb1dbca5f0dd3d214e111dd4848979911abc8e0e69ccc9
3
  size 293813988
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:87cca7363a760bda62b387b1f6b3a53ac8ca1563ad3ec20b995d07e94e342823
3
  size 293813988