Hugging Face's logo

Spaces:
MCG-NJU
/
PixNerd
Running on Zero

App Files Community
2
wangshuai6 commited on
Commit
bdce58d
·
1 Parent(s): e321f58

init

Browse files
Files changed (4) hide show
  1. configs_t2i/inference_heavydecoder.yaml +1 -1
  2. configs_t2i/pretraining_res256.yaml +1 -1
  3. configs_t2i/pretraining_res512.yaml +1 -1
  4. configs_t2i/sft_res512.yaml +1 -1
configs_t2i/inference_heavydecoder.yaml CHANGED
@@ -41,7 +41,7 @@ model:
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
- class_path: src.models.transformer.pixnerd_heavydecoder.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
 
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
+ class_path: src.models.transformer.pixnerd_t2i_heavydecoder.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
configs_t2i/pretraining_res256.yaml CHANGED
@@ -41,7 +41,7 @@ model:
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
- class_path: src.models.transformer.pixnerd.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
 
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
+ class_path: src.models.transformer.pixnerd_t2i.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
configs_t2i/pretraining_res512.yaml CHANGED
@@ -41,7 +41,7 @@ model:
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
- class_path: src.models.transformer.pixnerd.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
 
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
+ class_path: src.models.transformer.pixnerd_t2i.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
configs_t2i/sft_res512.yaml CHANGED
@@ -41,7 +41,7 @@ model:
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
- class_path: src.models.transformer.pixnerd.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16
 
41
  vae:
42
  class_path: src.models.autoencoder.pixel.PixelAE
43
  denoiser:
44
+ class_path: src.models.transformer.pixnerd_t2i.PixNerDiT
45
  init_args:
46
  in_channels: 3
47
  patch_size: 16