API: Fix loading

This commit is contained in:
somebody
2023-07-19 11:52:39 -05:00
parent 5f2600d338
commit fef42a6273
4 changed files with 31 additions and 6 deletions

View File

@@ -89,7 +89,10 @@ class HFTorchInferenceModel(HFInferenceModel):
self.hf_torch = True
self.lazy_load = True
self.low_mem = False
# TODO: Mayyyybe only keep one of these variables
self.nobreakmodel = False
self.breakmodel = False
self.post_token_hooks = [
PostTokenHooks.stream_tokens,