Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[Export Refactor] Prepare the module to be more general (before including transformers) #1908

Merged
merged 16 commits into from
Dec 19, 2023
Merged

[Export Refactor] Prepare the module to be more general (before including transformers) #1908

merged 16 commits into from
Dec 19, 2023

Commits on Dec 15, 2023

  1. adapt the export script to handle transformers

    @dbogunowicz
    dbogunowicz committed Dec 15, 2023
    Configuration menu
    Copy the full SHA
    23e892f View commit details
    Browse the repository at this point in the history

  2. Update src/sparseml/pytorch/image_classification/integration_helper_f… 

    …unctions.py
    @dbogunowicz
    dbogunowicz committed Dec 15, 2023
    Configuration menu
    Copy the full SHA
    2bb1f20 View commit details
    Browse the repository at this point in the history

  3. Delete tests/sparseml/export/transformers/__init__.py

    @dbogunowicz
    dbogunowicz committed Dec 15, 2023
    Configuration menu
    Copy the full SHA
    fb20ae2 View commit details
    Browse the repository at this point in the history

  4. Delete tests/sparseml/export/transformers/test_generative_transformer… 

    …s.py
    @dbogunowicz
    dbogunowicz committed Dec 15, 2023
    Configuration menu
    Copy the full SHA
    eab9fb1 View commit details
    Browse the repository at this point in the history

  5. Delete tests/sparseml/export/transformers/test_transformers.py

    @dbogunowicz
    dbogunowicz committed Dec 15, 2023
    Configuration menu
    Copy the full SHA
    0897615 View commit details
    Browse the repository at this point in the history

Commits on Dec 18, 2023

  1. Update src/sparseml/export/export.py 

    Co-authored-by: Benjamin Fineran <bfineran@users.noreply.github.com>
    @dbogunowicz @bfineran
    dbogunowicz and bfineran committed Dec 18, 2023
    Configuration menu
    Copy the full SHA
    9c7fac5 View commit details
    Browse the repository at this point in the history

  2. addressing review comments

    @dbogunowicz
    dbogunowicz committed Dec 18, 2023
    Configuration menu
    Copy the full SHA
    05a9ee3 View commit details
    Browse the repository at this point in the history

  3. [Export Refactor] Export transformers (#1909) 

    * cleanup

* Delete src/sparseml/transformers/integration_helper_functions_generative.py

* Delete src/sparseml/transformers/utils/optimizations.py

* Delete tests/sparseml/export/transformers/test_generative_transformers.py

* Delete tests/sparseml/transformers/test_integration_helper_functions_generative.py

* addressing PR reviews

* [Export Refactor] Export generative transformers(#1910)
    @dbogunowicz
    dbogunowicz committed Dec 18, 2023
    Configuration menu
    Copy the full SHA
    450b286 View commit details
    Browse the repository at this point in the history

Commits on Dec 19, 2023

  1. make tests green, remove using task to resolve the integration type

    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    313ecf8 View commit details
    Browse the repository at this point in the history

  2. fix all the tests after the merge, make integration resolution indepe… 

    …ndent of the task name
    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    6829514 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'feature/damian/export_adapt' of github.com:neuralmagic/… 

    …sparseml into feature/damian/export_adapt
    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    4ada09a View commit details
    Browse the repository at this point in the history

  4. fold generative transformers into transformer helper functions

    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    5840eb5 View commit details
    Browse the repository at this point in the history

  5. complete tests for export_data.py

    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    d7c2375 View commit details
    Browse the repository at this point in the history

  6. Update src/sparseml/export/export.py

    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    c308949 View commit details
    Browse the repository at this point in the history

  7. add tests that confirms that kv cache injection has been added

    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    ad2a7f0 View commit details
    Browse the repository at this point in the history

  8. move applying optimizations into integration helper functions

    @dbogunowicz
    dbogunowicz committed Dec 19, 2023
    Configuration menu
    Copy the full SHA
    9fbead1 View commit details
    Browse the repository at this point in the history