fix init.

huggingface · Dec 4, 2023 · 59df524 · 59df524
1 parent fa7bfe1
commit 59df524
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/benchmarks/base_classes.py b/benchmarks/base_classes.py
@@ -183,12 +183,21 @@ class ControlNetSDXLBenchmark(ControlNetBenchmark):
     pipeline_class = StableDiffusionXLControlNetPipeline
     root_ckpt = "stabilityai/stable-diffusion-xl-base-1.0"
 
+    def __init__(self, args):
+        super().__init__(args)
+
 
 class T2IAdapterBenchmark(ControlNetBenchmark):
     pipeline_class = StableDiffusionAdapterPipeline
     root_ckpt = "CompVis/stable-diffusion-v1-4"
 
+    def __init__(self, args):
+        super().__init__(args)
+
 
 class T2IAdapterSDXLBenchmark(T2IAdapterBenchmark):
     pipeline_class = StableDiffusionXLAdapterPipeline
     root_ckpt = "stabilityai/stable-diffusion-xl-base-1.0"
+
+    def __init__(self, args):
+        super().__init__(args)