mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-06-05 21:59:24 +02:00
API: Fix loading
This commit is contained in:
@@ -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,
|
||||
|
Reference in New Issue
Block a user