ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 25 days ago

Commit

0fa0525

·

verified ·

1 Parent(s): a9bc844

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -68,7 +68,7 @@
     "d_model": 512,
     "nhead": 8,
     "num_encoder_layers": 4,
-    "num_decoder_layers": 6,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 128,
     "down_dims": [

     "d_model": 512,
     "nhead": 8,
     "num_encoder_layers": 4,
+    "num_decoder_layers": 10,
     "dim_feedforward": 512,
     "diffusion_step_embed_dim": 128,
     "down_dims": [