From 6d79ca842403f9d54b84e419a99f0d10539a83b2 Mon Sep 17 00:00:00 2001 From: Sara Adkins Date: Fri, 30 Aug 2024 04:41:17 +0000 Subject: [PATCH] remove recipe app --- .../transformers/sparsification/sparse_model.py | 10 +--------- 1 file changed, 1 insertion(+), 9 deletions(-) diff --git a/src/llmcompressor/transformers/sparsification/sparse_model.py b/src/llmcompressor/transformers/sparsification/sparse_model.py index 3af7d2b4d..4153ec4f4 100644 --- a/src/llmcompressor/transformers/sparsification/sparse_model.py +++ b/src/llmcompressor/transformers/sparsification/sparse_model.py @@ -14,14 +14,10 @@ from torch.nn import Module from transformers import AutoModelForCausalLM, PreTrainedModel -from llmcompressor.pytorch.model_load.helpers import initialize_recipe from llmcompressor.transformers.sparsification.compressed_tensors_utils import ( modify_save_pretrained, ) -from llmcompressor.transformers.utils.helpers import ( - download_model_directory, - resolve_recipe, -) +from llmcompressor.transformers.utils.helpers import download_model_directory __all__ = ["SparseAutoModel", "SparseAutoModelForCausalLM", "get_shared_tokenizer_src"] @@ -143,10 +139,6 @@ def skip(*args, **kwargs): model_path=pretrained_model_name_or_path, model=model ) - recipe = resolve_recipe(recipe=recipe, model_path=pretrained_model_name_or_path) - if recipe: - initialize_recipe(model=model, recipe_path=recipe) - return model