Fix for missing import

Fix for model name being a path which caused save issues
This commit is contained in:
ebolam
2023-05-20 11:01:49 -04:00
parent 19559d5eef
commit 513b8575e7
2 changed files with 2 additions and 1 deletions

View File

@@ -8,6 +8,7 @@ import koboldai_settings
from logger import logger
from modeling.inference_model import InferenceModel
import torch
import gc
class HFInferenceModel(InferenceModel):