Commit bc6434b
[Bugfix] Fix tests, Add Model Free Readme (#2119)
## Purpose ##
* Fix failing tests
* Add readme for model free ptq source code
## Changes ##
* Use `initialize` instead of `on_initialize` so that the
`Modifier.initialized_` flag is set
* Remove expectation that lm_head inputs are on any particular device
## Testing ##
* Replicated test failures locally, confirmed that changes fix tests
---------
Signed-off-by: Kyle Sayers <kylesayrs@gmail.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>1 parent 03e694a commit bc6434b
File tree
3 files changed
+10
-3
lines changed- src/llmcompressor/entrypoints/model_free
- tests/llmcompressor
- modifiers/transform
- utils
3 files changed
+10
-3
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
| 1 | + | |
| 2 | + | |
| 3 | + | |
| 4 | + | |
| 5 | + | |
| 6 | + | |
| 7 | + | |
| 8 | + | |
| 9 | + | |
Lines changed: 1 addition & 2 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
45 | 45 | | |
46 | 46 | | |
47 | 47 | | |
48 | | - | |
49 | | - | |
| 48 | + | |
50 | 49 | | |
51 | 50 | | |
52 | 51 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
174 | 174 | | |
175 | 175 | | |
176 | 176 | | |
177 | | - | |
178 | 177 | | |
0 commit comments