From 9dee9b5c6db4ac6bd0c91a4336c84e360c840903 Mon Sep 17 00:00:00 2001 From: henk717 Date: Wed, 9 Mar 2022 12:03:37 +0100 Subject: [PATCH] Ignore incorrect problems --- aiserver.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/aiserver.py b/aiserver.py index ec69fbc2..53fb5b87 100644 --- a/aiserver.py +++ b/aiserver.py @@ -458,9 +458,9 @@ def move_model_to_devices(model): gc.collect() GPTNeoModel.forward = breakmodel.new_forward_neo if("GPTJModel" in globals()): - GPTJModel.forward = breakmodel.new_forward_neo + GPTJModel.forward = breakmodel.new_forward_neo # type: ignore if("XGLMModel" in globals()): - XGLMModel.forward = breakmodel.new_forward_xglm + XGLMModel.forward = breakmodel.new_forward_xglm # type: ignore generator = model.generate if(hasattr(model, "transformer")): breakmodel.move_hidden_layers(model.transformer)