Update hf-diffusers.libsonnet

Copy from #1038 to r2.2
GoogleCloudPlatform · Jan 9, 2024 · afe10f3 · afe10f3
1 parent 5aed632
commit afe10f3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/pytorch/r2.2/hf-diffusers.libsonnet b/tests/pytorch/r2.2/hf-diffusers.libsonnet
@@ -22,7 +22,8 @@ local tpus = import 'templates/tpus.libsonnet';
       '--lr_scheduler=constant',
       '--lr_warmup_steps=0',
       '--output_dir=/tmp/sd-pokemon-model',
-      '--checkpoints_total_limit=3',
+      '--checkpoints_total_limit=1',
+      '--checkpointing_steps=6000',
     ],
   },