Skip to content

Commit

Permalink
Update integration_helper_functions.py
Browse files Browse the repository at this point in the history
  • Loading branch information
robertgshaw2-redhat authored Mar 23, 2024
1 parent 3196bea commit 95888f6
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions src/sparseml/transformers/integration_helper_functions.py
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@
from sparseml.transformers.utils.load_task_dataset import load_task_dataset
from sparseml.transformers.utils.optimizations import apply_kv_cache_injection


_LOGGER = logging.getLogger(__name__)


Expand Down

0 comments on commit 95888f6

Please sign in to comment.