mirror of
https://github.com/KoboldAI/KoboldAI-Client.git
synced 2025-03-09 16:10:16 +01:00
Fix for loading models on CPU only that don't support breakmodel
This commit is contained in:
parent
edd50fc809
commit
0311cc215e
@ -1514,8 +1514,10 @@ def get_model_info(model, directory=""):
|
||||
pass
|
||||
else:
|
||||
layer_count = get_layer_count(model, directory=directory)
|
||||
print("Layer Count: {}".format(layer_count))
|
||||
if layer_count is None:
|
||||
breakmodel = False
|
||||
gpu = True
|
||||
else:
|
||||
breakmodel = True
|
||||
if model in ["NeoCustom", "GPT2Custom"]:
|
||||
|
Loading…
x
Reference in New Issue
Block a user