From a000e3d05cfa58d7ee473c527697ea54b0a3646e Mon Sep 17 00:00:00 2001 From: Adrian Tamas Date: Tue, 25 Feb 2025 11:18:45 +0000 Subject: [PATCH] Fix: Ensure install_diffusers() is called before HUGGING FACE token check --- torchbenchmark/canary_models/stable_diffusion_xl/install.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchbenchmark/canary_models/stable_diffusion_xl/install.py b/torchbenchmark/canary_models/stable_diffusion_xl/install.py index 993f3043c8..df3a68308c 100644 --- a/torchbenchmark/canary_models/stable_diffusion_xl/install.py +++ b/torchbenchmark/canary_models/stable_diffusion_xl/install.py @@ -16,10 +16,10 @@ def load_model_checkpoint(): if __name__ == "__main__": + install_diffusers() if not "HUGGING_FACE_HUB_TOKEN" in os.environ: warnings.warn( "Make sure to set `HUGGINGFACE_HUB_TOKEN` so you can download weights" ) else: - install_diffusers() load_model_checkpoint()