Skip saving pretrained model in diffuser nightly test (GoogleCloudPla…

…tform#984) * Skip saving pretrained model for diffuser, in nightly
wonjoolee95 · Sep 6, 2023 · 0ec0af9 · 0ec0af9
1 parent a65ecdd
commit 0ec0af9
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/pytorch/nightly/hf-diffusers.libsonnet b/tests/pytorch/nightly/hf-diffusers.libsonnet
@@ -55,6 +55,9 @@ local tpus = import 'templates/tpus.libsonnet';
         sed '/accelerate/d' requirements.txt > clean_requirements.txt
         sed '/torchvision/d' requirements.txt > clean_requirements.txt
         pip install -r clean_requirements.txt
+
+        # Skip saving the pretrained model, which contains invalid tensor storage
+        sed -i 's/pipeline.save_pretrained(args.output_dir)//g' train_text_to_image.py
       |||,
     },
   },