Quick fix for quantization/custom op list loading (#1657)

Signed-off-by: Daniel Socek <[email protected]>
huggingface · Dec 23, 2024 · f2a9079 · f2a9079
1 parent f494e51
commit f2a9079
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/examples/stable-diffusion/text_to_image_generation.py b/examples/stable-diffusion/text_to_image_generation.py
@@ -440,9 +440,11 @@ def main():
 
     kwargs_call["quant_mode"] = args.quant_mode
 
-    # Instantiate a Stable Diffusion pipeline class
-    import habana_frameworks.torch.core as htcore  # noqa: F401
+    if args.quant_mode != "disable":
+        # Import htcore here to support model quantization
+        import habana_frameworks.torch.core as htcore  # noqa: F401
 
+    # Instantiate a Stable Diffusion pipeline class
     if sdxl:
         # SDXL pipelines
         if controlnet: