Update __init__.py

pytorch · Jul 13, 2023 · b9b6ca8 · b9b6ca8
1 parent b39b098
commit b9b6ca8
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
@@ -5,6 +5,7 @@
 """
 from torchbenchmark.tasks import COMPUTER_VISION
 from torchbenchmark.util.model import BenchmarkModel
+from torchbenchmark.util.framework.huggingface.model_factory import HuggingFaceAuthMixin
 
 import torch
 from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler