helloworld-S commited on
Commit
07a007b
·
verified ·
1 Parent(s): c615b4c

Update src/flux/pipeline_tools.py

Browse files
Files changed (1) hide show
  1. src/flux/pipeline_tools.py +3 -3
src/flux/pipeline_tools.py CHANGED
@@ -31,9 +31,9 @@ from diffusers.models.transformers.transformer_flux import (
31
  )
32
  from torchvision.transforms import ToPILImage
33
  from peft.tuners.tuners_utils import BaseTunerLayer
34
- from optimum.quanto import (
35
- freeze, quantize, QTensor, qfloat8, qint8, qint4, qint2,
36
- )
37
  import re
38
  import safetensors
39
  from src.adapters.mod_adapters import CLIPModAdapter
 
31
  )
32
  from torchvision.transforms import ToPILImage
33
  from peft.tuners.tuners_utils import BaseTunerLayer
34
+ # from optimum.quanto import (
35
+ # freeze, quantize, QTensor, qfloat8, qint8, qint4, qint2,
36
+ # )
37
  import re
38
  import safetensors
39
  from src.adapters.mod_adapters import CLIPModAdapter