Skip to content

Commit

Permalink
remove recipe app
Browse files Browse the repository at this point in the history
  • Loading branch information
Sara Adkins committed Aug 30, 2024
1 parent e33933e commit 6d79ca8
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions src/llmcompressor/transformers/sparsification/sparse_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"]

Expand Down Expand Up @@ -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


Expand Down

0 comments on commit 6d79ca8

Please # to comment.