more accurate readme

lhmzll · Jul 10, 2022 · 4173e88 · 4173e88
1 parent 3dae43f
commit 4173e88
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -355,7 +355,8 @@ prior_network = DiffusionPriorNetwork(
 diffusion_prior = DiffusionPrior(
     net = prior_network,
     clip = clip,
-    timesteps = 100,
+    timesteps = 1000,
+    sample_timesteps = 64,
     cond_drop_prob = 0.2
 ).cuda()