InventorsHub commited on
Commit
96befc7
·
verified ·
1 Parent(s): 92e17b0

Update safety_module.py

Browse files
Files changed (1) hide show
  1. safety_module.py +1 -1
safety_module.py CHANGED
@@ -22,7 +22,7 @@ import functools
22
  # )
23
 
24
  @functools.lru_cache(maxsize=1)
25
- @spaces.GPU
26
  def llm_gpu():
27
  model_path = hf_hub_download(
28
  repo_id="Inventors-Hub/SwarmChat-models",
 
22
  # )
23
 
24
  @functools.lru_cache(maxsize=1)
25
+ @spaces.CPU
26
  def llm_gpu():
27
  model_path = hf_hub_download(
28
  repo_id="Inventors-Hub/SwarmChat-models",