rixty/KoboldAI-Client
1
1
Fork 0
mirror of https://github.com/KoboldAI/KoboldAI-Client.git synced 2025-06-05 21:59:24 +02:00
Code Issues Projects Releases Wiki Activity

Ignore incorrect problems

Browse Source
This commit is contained in:
henk717
2022-03-09 12:03:37 +01:00
parent a28e553412
commit 9dee9b5c6d
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
aiserver.py
View File

@ -458,9 +458,9 @@ def move_model_to_devices(model):
gc.collect() gc.collect()
GPTNeoModel.forward = breakmodel.new_forward_neo GPTNeoModel.forward = breakmodel.new_forward_neo
if("GPTJModel" in globals()): if("GPTJModel" in globals()):
GPTJModel.forward = breakmodel.new_forward_neo GPTJModel.forward = breakmodel.new_forward_neo # type: ignore
if("XGLMModel" in globals()): if("XGLMModel" in globals()):
XGLMModel.forward = breakmodel.new_forward_xglm XGLMModel.forward = breakmodel.new_forward_xglm # type: ignore
generator = model.generate generator = model.generate
if(hasattr(model, "transformer")): if(hasattr(model, "transformer")):
breakmodel.move_hidden_layers(model.transformer) breakmodel.move_hidden_layers(model.transformer)