'undefined'

2024-04-23 01:34:50 +08:00
parent f8b9dc0c1d
commit 6f5523ddbb
1 changed files with 1 additions and 0 deletions
@@ -146,5 +146,6 @@ def query_beautify_prompt_text(model_name, n_gpu_layers, text, style_presets, do
        return full_response

    except Exception as e:
+        mz_llama_cpp.freed_gpu_memory(model_file=model_file)
        # mz_utils.Utils.print_log(f"Error in auto_prompt_text: {e}")
        raise e