Update hf-diffusers.libsonnet

GoogleCloudPlatform · Oct 3, 2023 · 0ffda5f · 0ffda5f
1 parent c99553c
commit 0ffda5f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/pytorch/r2.1/hf-diffusers.libsonnet b/tests/pytorch/r2.1/hf-diffusers.libsonnet
@@ -24,6 +24,7 @@ local tpus = import 'templates/tpus.libsonnet';
       '--lr_warmup_steps=0',
       '--output_dir=/tmp/sd-pokemon-model',
       '--checkpoints_total_limit=3',
+      '--checkpointing_steps=6000',
     ],
   },