Fix wildcard import

danielz02 · Mar 26, 2024 · fba2378 · fba2378
1 parent d8da4ca
commit fba2378
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/helm/proxy/clients/huggingface_client.py b/src/helm/proxy/clients/huggingface_client.py
@@ -91,8 +91,8 @@ def __init__(self, model_config: HuggingFaceModelConfig):
                 hlog(self.model.hf_device_map)
             elif quantization_config and quantization_config.model_loader == ModelLoader.SQUEEZELLM:
                 import torch
-                from squeezellm.modelutils import *
-                from squeezellm.quant import *
+                from squeezellm.modelutils import find_layers
+                from squeezellm.quant import make_quant_lut
 
                 if (
                         "xgen" in quantization_config.quant_file