Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00: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
parent 38c2e7801f
commit e41025ddc7
4 changed files with 2 additions and 2 deletions
Show Stats 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