Skip to content

GitLab

Switch branch/tag
  1. 22 Oct, 2024 1 commit
    • Leonid Sinev's avatar
      [Fix] Replace generic exception classes with a more specific ones (#1989) · d4ae9635
      Leonid Sinev authored 
      * Replace generic exception classes with a more specific ones

* rerun pre-commit to pass linter tests

* Revert "rerun pre-commit to pass linter tests"

This reverts commit 67f88ccf144469853217704520e613196042d859.

* reduce repetitions in errors or so

* Replace generic exception class with a more specific one
      d4ae9635
  3. 28 Feb, 2024 1 commit
  5. 14 Feb, 2024 1 commit
  7. 22 Jan, 2024 1 commit
  9. 22 Dec, 2023 1 commit
    • Hailey Schoelkopf's avatar
      Upstream Mamba Support (`mamba_ssm`) (#1110) · 5503b274
      Hailey Schoelkopf authored 
      * modularize HFLM code

* pass through extra kwargs to AutoModel.from_pretrained call

* remove explicit model_kwargs

* rename gptq -> autogptq

* fix tokenizer pad token errors

* ensure model always respects device_map and autogptq's selected devices

* add a _get_config helper fn

* add mambaLMWrapper

* add mamba extra

* add mamba extra

* fix conditional import

* Fix botched merge commit

* Remove beginning-of-file comment for consistency

* Add docstring for mambaLM re: supported kwargs

* Alphabetize extras

* Update extras table

* appease precommit

* run precommit on mamba_lm
      5503b274