From 21ecc3701ef1b456278ff26998e4ada4d2b3eca1 Mon Sep 17 00:00:00 2001 From: oobabooga <112222186+oobabooga@users.noreply.github.com> Date: Wed, 17 May 2023 11:23:13 -0300 Subject: [PATCH] Avoid a name conflict --- modules/models.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/modules/models.py b/modules/models.py index 41d4f479..c63ead75 100644 --- a/modules/models.py +++ b/modules/models.py @@ -257,17 +257,17 @@ def GPTQ_loader(model_name): # No monkey patch else: - from modules.GPTQ_loader import load_quantized + import modules.GPTQ_loader - model = load_quantized(model_name) + model = modules.GPTQ_loader.load_quantized(model_name) return model def AutoGPTQ_loader(model_name): - from modules.AutoGPTQ_loader import load_quantized + import modules.AutoGPTQ_loader - return load_quantized(model_name) + return modules.AutoGPTQ_loader.load_quantized(model_name) def get_max_memory_dict():