Update safety_module.py

safety_module.py

@@ -1,6 +1,6 @@
 from llama_cpp import Llama
 from huggingface_hub import hf_hub_download
-import spaces
+# import spaces
 import functools
 
 # Download only the safety model shard from the model repo
@@ -21,8 +21,8 @@ import functools
 #     use_mlock=False,
 # )
 
+# @spaces.CPU
 @functools.lru_cache(maxsize=1)
-@spaces.CPU
 def llm_gpu():
     model_path = hf_hub_download(
         repo_id="Inventors-Hub/SwarmChat-models",