From c1809a3380edc10edb93ed8b9bb29744b9ec5fcf Mon Sep 17 00:00:00 2001
From: Yu xing <xyyimian@gmail.com>
Date: Wed, 21 Aug 2024 16:49:44 -0700
Subject: [PATCH] around it now

---
 nexa/gguf/nexa_inference_image.py  | 2 +-
 nexa/gguf/nexa_inference_text.py   | 2 +-
 nexa/gguf/nexa_inference_vlm.py    | 2 +-
 nexa/gguf/nexa_inference_voice.py  | 2 +-
 tests/{test_vlm.py => vlm_test.py} | 0
 5 files changed, 4 insertions(+), 4 deletions(-)
 rename tests/{test_vlm.py => vlm_test.py} (100%)

diff --git a/nexa/gguf/nexa_inference_image.py b/nexa/gguf/nexa_inference_image.py
index 279923a0..d9d38bfa 100644
--- a/nexa/gguf/nexa_inference_image.py
+++ b/nexa/gguf/nexa_inference_image.py
@@ -81,7 +81,7 @@ def __init__(self, model_path, **kwargs):
                 logging.error("Failed to load the model or pipeline.")
                 exit(1)
 
-    #@SpinningCursorAnimation()
+    @SpinningCursorAnimation()
     def _load_model(self, model_path: str):
         with suppress_stdout_stderr():
             from nexa.gguf.sd.stable_diffusion import StableDiffusion
diff --git a/nexa/gguf/nexa_inference_text.py b/nexa/gguf/nexa_inference_text.py
index 7395093e..c93d3519 100644
--- a/nexa/gguf/nexa_inference_text.py
+++ b/nexa/gguf/nexa_inference_text.py
@@ -98,7 +98,7 @@ def create_embedding(
         """
         return self.model.create_embedding(input)
 
-    #@SpinningCursorAnimation()
+    @SpinningCursorAnimation()
     def _load_model(self):
         logging.debug(f"Loading model from {self.downloaded_path}")
         start_time = time.time()
diff --git a/nexa/gguf/nexa_inference_vlm.py b/nexa/gguf/nexa_inference_vlm.py
index 63b8f091..47e54786 100644
--- a/nexa/gguf/nexa_inference_vlm.py
+++ b/nexa/gguf/nexa_inference_vlm.py
@@ -146,7 +146,7 @@ def __init__(self, model_path, stop_words=None, **kwargs):
                 )
                 exit(1)
 
-    #@SpinningCursorAnimation()
+    @SpinningCursorAnimation()
     def _load_model(self):
         logging.debug(f"Loading model from {self.downloaded_path}")
         start_time = time.time()
diff --git a/nexa/gguf/nexa_inference_voice.py b/nexa/gguf/nexa_inference_voice.py
index 92eeaee5..b6437442 100644
--- a/nexa/gguf/nexa_inference_voice.py
+++ b/nexa/gguf/nexa_inference_voice.py
@@ -69,7 +69,7 @@ def __init__(self, model_path, **kwargs):
                 exit(1)
 
 
-    #@SpinningCursorAnimation()
+    @SpinningCursorAnimation()
     def _load_model(self):
         from faster_whisper import WhisperModel
 
diff --git a/tests/test_vlm.py b/tests/vlm_test.py
similarity index 100%
rename from tests/test_vlm.py
rename to tests/vlm_test.py