Move requantize.py to the quatnization/ dir.

invoke-ai · Aug 21, 2024 · e41025d · e41025d
1 parent 38c2e78
commit e41025d
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 2 deletions.
diff --git a/invokeai/backend/quantization/__init__.py b/invokeai/backend/quantization/__init__.py
diff --git a/invokeai/backend/quantization/fast_quantized_diffusion_model.py b/invokeai/backend/quantization/fast_quantized_diffusion_model.py
@@ -14,7 +14,7 @@
 from optimum.quanto.models import QuantizedDiffusersModel
 from optimum.quanto.models.shared_dict import ShardedStateDict
 
-from invokeai.backend.requantize import requantize
+from invokeai.backend.quantization.requantize import requantize
 
 
 class FastQuantizedDiffusersModel(QuantizedDiffusersModel):

diff --git a/invokeai/backend/quantization/fast_quantized_transformers_model.py b/invokeai/backend/quantization/fast_quantized_transformers_model.py
@@ -9,7 +9,7 @@
 from transformers.models.auto import AutoModelForTextEncoding
 from transformers.utils import SAFE_WEIGHTS_INDEX_NAME, SAFE_WEIGHTS_NAME, is_accelerate_available
 
-from invokeai.backend.requantize import requantize
+from invokeai.backend.quantization.requantize import requantize
 
 
 class FastQuantizedTransformersModel(QuantizedTransformersModel):

diff --git a/invokeai/backend/requantize.py → invokeai/backend/quantization/requantize.py b/invokeai/backend/requantize.py → invokeai/backend/quantization/requantize.py