From 0011328dea61964e77c737b0e7c183363868640f Mon Sep 17 00:00:00 2001 From: Ean Garvey Date: Tue, 9 Apr 2024 10:50:08 -0500 Subject: [PATCH] cleanup pipeline test artifacts after completion. --- models/turbine_models/tests/sdxl_test.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/models/turbine_models/tests/sdxl_test.py b/models/turbine_models/tests/sdxl_test.py index ad838061b..3b351c6f3 100644 --- a/models/turbine_models/tests/sdxl_test.py +++ b/models/turbine_models/tests/sdxl_test.py @@ -581,6 +581,18 @@ def test05_t2i_generate_images(self): arguments["seed"], ) print("Image generation complete.") + os.remove(os.path.join(arguments["pipeline_dir"], "prompt_encoder.vmfb")) + os.remove( + os.path.join( + arguments["pipeline_dir"], + arguments["scheduler_id"] + + "_unet_" + + str(arguments["num_inference_steps"]) + + ".vmfb", + ) + ) + os.remove(os.path.join(arguments["pipeline_dir"], "vae_decode.vmfb")) + os.remove(os.path.join(arguments["pipeline_dir"], "full_pipeline.vmfb")) if __name__ == "__main__":