From 513b8575e71d164fc82747009f8fd3391f4ceb28 Mon Sep 17 00:00:00 2001
From: ebolam <ebolam@gmail.com>
Date: Sat, 20 May 2023 11:01:49 -0400
Subject: [PATCH] Fix for missing import Fix for model name being a path which
 caused save issues

---
 aiserver.py                     | 2 +-
 modeling/inference_models/hf.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/aiserver.py b/aiserver.py
index a8591dc3..38ffc3f6 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -6165,7 +6165,7 @@ def UI_2_select_model(data):
                     else:
                         logger.debug("{} says invalid".format(model_backend))
                     
-                output.append({'label': path[1], 'name': path[0], 'size': "", "menu": "Custom", 'path': path[0], 'isMenu': not valid})
+                output.append({'label': path[1], 'name': path[1], 'size': "", "menu": "Custom", 'path': path[0], 'isMenu': not valid})
             emit("open_model_load_menu", {"items": output+[{'label': 'Return to Main Menu', 'name':'mainmenu', 'size': "", "menu": "Custom", 'isMenu': True}], 'breadcrumbs': breadcrumbs})            
     return
 
diff --git a/modeling/inference_models/hf.py b/modeling/inference_models/hf.py
index eff3d1ce..318423d5 100644
--- a/modeling/inference_models/hf.py
+++ b/modeling/inference_models/hf.py
@@ -8,6 +8,7 @@ import koboldai_settings
 from logger import logger
 from modeling.inference_model import InferenceModel
 import torch
+import gc
 
 
 class HFInferenceModel(InferenceModel):