Update modules/load_model.py

modules/load_model.py

@@ -1,3 +1,5 @@
+import spaces
+
 import os
 import gradio as gr
 import copy
@@ -7,6 +9,7 @@ import random
 from huggingface_hub import hf_hub_download  
 import time
 
+@spaces.GPU
 def load_model(path, n_ctx, n_gpu_layers, n_threads, verbose, f16_kv, logits_all, vocab_only, use_mmap, use_mlock, n_batch, last_n_tokens_size, low_vram, rope_freq_base, rope_freq_scale):
     try:
         dir = os.getcwd()