ISdept
/

transformer-diffusion

Model card Files Files and versions

ISdept commited on 1 day ago

Commit

9984960

·

verified ·

1 Parent(s): 1496f74

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -73,5 +73,6 @@
     "optimizer_eps": 1e-08,
     "optimizer_weight_decay": 1e-06,
     "scheduler_name": "cosine",
-    "scheduler_warmup_steps": 500
 }

     "optimizer_eps": 1e-08,
     "optimizer_weight_decay": 1e-06,
     "scheduler_name": "cosine",
+    "scheduler_warmup_steps": 500,
+    "num_inference_steps": 100
 }