namin72
/

diffusion_brown_capybara

Model card Files Files and versions

namin72 commited on Jan 25

Commit

2749306

·

verified ·

1 Parent(s): a76e60d

Update config.json

Files changed (1) hide show

config.json +2 -1

config.json CHANGED Viewed

@@ -86,5 +86,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,
+    "use_vae": true
 }