diff --git a/modules/model_flux.py b/modules/model_flux.py index e3adf3981..b89714d91 100644 --- a/modules/model_flux.py +++ b/modules/model_flux.py @@ -213,7 +213,6 @@ def load_transformer(file_path): # triggered by opts.sd_unet change if _transformer is not None: transformer = _transformer elif quant == 'fp8' or quant == 'fp4' or quant == 'nf4' or 'Model' in shared.opts.bnb_quantization: - print('HERE0') _transformer, _text_encoder_2 = load_flux_bnb(file_path, diffusers_load_config) if _transformer is not None: transformer = _transformer @@ -223,7 +222,6 @@ def load_transformer(file_path): # triggered by opts.sd_unet change if _transformer is not None: transformer = _transformer else: - print('HERE1') quant_args = {} quant_args = model_quant.create_bnb_config(quant_args) if quant_args: @@ -232,8 +230,6 @@ def load_transformer(file_path): # triggered by opts.sd_unet change quant_args = model_quant.create_ao_config(quant_args) if quant_args: model_quant.load_torchao(f'Load model: type=Sana quant={quant_args}') - print('HERE2', diffusers_load_config) - print('HERE3', quant_args) transformer = diffusers.FluxTransformer2DModel.from_single_file(file_path, **diffusers_load_config, **quant_args) if transformer is None: shared.log.error('Failed to load UNet model')