Enhance load_empty_model import (#1930)

Signed-off-by: Kaihui-intel <[email protected]>

Author: Kaihui-intel

neural_compressor/torch/__init__.py

@@ -11,3 +11,4 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
+from .utils import load_empty_model

test/3x/torch/quantization/weight_only/test_gptq.py

@@ -183,7 +183,7 @@ def test_layer_wise(self):
         model = convert(model)
         q_label = model(self.example_inputs)[0]
 
-        from neural_compressor.torch.utils import load_empty_model
+        from neural_compressor.torch import load_empty_model
 
         model = load_empty_model("hf-internal-testing/tiny-random-GPTJForCausalLM")
 

test/3x/torch/quantization/weight_only/test_rtn.py

@@ -167,7 +167,7 @@ def test_quant_lm_head(self):
         ), "The tied lm_head weight is not deep copied, please check!"
 
     def test_layer_wise(self):
-        from neural_compressor.torch.utils import load_empty_model
+        from neural_compressor.torch import load_empty_model
 
         model = load_empty_model("hf-internal-testing/tiny-random-GPTJForCausalLM")
         quant_config = RTNConfig(