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 quantization scripts to a scripts/ subdir.

Browse Source
This commit is contained in:
Ryan Dick 2024-08-23 18:08:37 +00:00 committed by Brandon
parent a21f6c4964
commit dff4a88baa
3 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/quantization/load_flux_model_bnb_llm_int8.py → invokeai/backend/quantization/scripts/load_flux_model_bnb_llm_int8.py
View File

@ -6,7 +6,7 @@ from flux.util import configs as flux_configs
from safetensors.torch import load_file, save_file from safetensors.torch import load_file, save_file
from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8 from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8
from invokeai.backend.quantization.load_flux_model_bnb_nf4 import log_time from invokeai.backend.quantization.scripts.load_flux_model_bnb_nf4 import log_time
def main(): def main():

0
invokeai/backend/quantization/load_flux_model_bnb_nf4.py → invokeai/backend/quantization/scripts/load_flux_model_bnb_nf4.py
View File

2
invokeai/backend/quantization/quantize_t5_xxl_bnb_llm_int8.py → invokeai/backend/quantization/scripts/quantize_t5_xxl_bnb_llm_int8.py
View File

@ -5,7 +5,7 @@ from safetensors.torch import load_file, save_file
from transformers import AutoConfig, AutoModelForTextEncoding, T5EncoderModel from transformers import AutoConfig, AutoModelForTextEncoding, T5EncoderModel
from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8 from invokeai.backend.quantization.bnb_llm_int8 import quantize_model_llm_int8
from invokeai.backend.quantization.load_flux_model_bnb_nf4 import log_time from invokeai.backend.quantization.scripts.load_flux_model_bnb_nf4 import log_time
def load_state_dict_into_t5(model: T5EncoderModel, state_dict: dict): def load_state_dict_into_t5(model: T5EncoderModel, state_dict: dict):