github/InvokeAI
1
1
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI.git synced 2026-02-07 12:15:06 -05:00
Code Issues Packages Projects Releases Wiki Activity

Move requantize.py to the quatnization/ dir.

Browse Source
This commit is contained in:
Ryan Dick
2024-08-21 18:21:05 +00:00
committed by Brandon
parent 1787de6836
commit cfac7c8189
4 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
invokeai/backend/quantization/__init__.py Normal file
View File

2
invokeai/backend/quantization/fast_quantized_diffusion_model.py
View File

@@ -14,7 +14,7 @@ from diffusers.utils import (
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):

2
invokeai/backend/quantization/fast_quantized_transformers_model.py
View File

@@ -9,7 +9,7 @@ from transformers.modeling_utils import get_checkpoint_shard_files, load_state_d
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):

0
invokeai/backend/requantize.py → invokeai/backend/quantization/requantize.py
View File