From 0ffda5fb8af7c4141a1201c7f4716a4351cd0aec Mon Sep 17 00:00:00 2001
From: Manfei <41607353+ManfeiBai@users.noreply.github.com>
Date: Mon, 2 Oct 2023 19:53:57 -0700
Subject: [PATCH] Update hf-diffusers.libsonnet

---
 tests/pytorch/r2.1/hf-diffusers.libsonnet | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/pytorch/r2.1/hf-diffusers.libsonnet b/tests/pytorch/r2.1/hf-diffusers.libsonnet
index 5141ff6ca..8bc7667a9 100644
--- 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',
     ],
   },