better handle any quant lib requirements

Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2024-10-12 13:36:09 -04:00
parent e2d5cd558b
commit ea0dfebe2d
25 changed files with 164 additions and 130 deletions
+2 -3
View File
@@ -4,7 +4,7 @@ from typing import Union
from diffusers import StableDiffusionPipeline, StableDiffusionXLPipeline, FluxPipeline, ControlNetModel
from modules.control.units import detect
from modules.shared import log, opts, listdir
from modules import errors, sd_models, devices
from modules import errors, sd_models, devices, model_quant
what = 'ControlNet'
@@ -229,8 +229,7 @@ class ControlNet():
elif "ControlNet" in opts.optimum_quanto_weights:
try:
log.debug(f'Control {what} model Optimum Quanto: id="{model_id}"')
from installer import install
install('optimum-quanto', quiet=True)
model_quant.load_quanto('Load model: type=ControlNet')
from modules.sd_models_compile import optimum_quanto_model
self.model = optimum_quanto_model(self.model)
except Exception as e: