Update __init__.py

pytorch · Jul 13, 2023 · 720a163 · 720a163
1 parent 3ae570b
commit 720a163
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torchbenchmark/models/stable_diffusion/__init__.py b/torchbenchmark/models/stable_diffusion/__init__.py
@@ -22,6 +22,7 @@ class Model(HuggingFaceAuthMixin, BenchmarkModel):
 
 
     def __init__(self, test, device, jit=False, batch_size=None, extra_args=[]):
+        HuggingFaceAuthMixin.__init__(self)
         super().__init__(test=test, device=device, jit=jit,
                          batch_size=batch_size, extra_args=extra_args)
         assert self.dargs.precision == "fp16", f"Stable Diffusion model only supports fp16 precision."