More cleaning

2025-06-05 21:59:24 +02:00 · 2023-05-28 13:22:32 -05:00
parent 14241fc156
commit ed0728188a
3 changed files with 48 additions and 11 deletions
--- a/modeling/inference_models/generic_hf_torch/class.py
+++ b/modeling/inference_models/generic_hf_torch/class.py
@@ -238,6 +238,7 @@ class model_backend(HFTorchInferenceModel):
                    shutil.rmtree("cache/")

        self.patch_embedding()
+        lazy_loader.post_load_cleanup()

        self.model.kai_model = self
        utils.koboldai_vars.modeldim = self.get_hidden_size()