YiYiXu
/

quant-block

Model card Files Files and versions

YiYiXu HF Staff commited on Jan 30

Commit

3475425

·

verified ·

1 Parent(s): 7739980

Update block.py

Files changed (1) hide show

block.py +2 -2

block.py CHANGED Viewed

@@ -2,12 +2,12 @@ from typing import List, Optional
 from diffusers.modular_pipelines import (
     InputParam,
     OutputParam,
-    ModularPipelineBlock,
     PipelineState,
 )
-class QuantizationConfigBlock(ModularPipelineBlock):
     """Block to create BitsAndBytes quantization config for model loading."""
     @property

 from diffusers.modular_pipelines import (
     InputParam,
     OutputParam,
+    ModularPipelineBlocks,
     PipelineState,
 )
+class QuantizationConfigBlock(ModularPipelineBlocks):
     """Block to create BitsAndBytes quantization config for model loading."""
     @property