Skip to content

Commit be1ed76

Browse files
authored
Fix slicegpt bug (microsoft#2175)
## Describe your changes import typo ## Checklist before requesting a review - [ ] Add unit tests for this change. - [ ] Make sure all tests can pass. - [ ] Update documents if necessary. - [ ] Lint and apply fixes to your code by running `lintrunner -a` - [ ] Is this a user-facing change? If yes, give a description of this change to be included in the release notes. ## (Optional) Issue link
1 parent 183703b commit be1ed76

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

olive/model/handler/pytorch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -174,7 +174,7 @@ def load_model(self, rank: int = None, cache_model: bool = True) -> "torch.nn.Mo
174174
return model
175175

176176
def _load_slicegpt_model(self):
177-
from slicgpt.hf_utils import load_sliced_model
177+
from slicegpt.hf_utils import load_sliced_model
178178

179179
model_name = self.model_attributes.get("model_name")
180180
if not model_name:

0 commit comments

Comments
 (0)