.

scripts/anime.py

@@ -19,6 +19,7 @@ import spaces
 model = None
 device = None
 
+@spaces.GPU
 def init_model(use_local=False):
     global model, device
     model_opt = "default"

scripts/generate_prompt.py

@@ -41,6 +41,7 @@ def download_model_files(repo_id, model_dir, sub_dir, files, sub_files):
     for file in sub_files:
         hf_hub_download(repo_id, file, subfolder=sub_dir, cache_dir=os.path.join(model_dir, sub_dir), force_download=True, force_filename=file)
 
+@spaces.GPU
 def load_wd14_tagger_model():
     """WD14タグ付けモデルをロード"""
     model_dir = "wd14_tagger_model"

scripts/process_utils.py

@@ -35,6 +35,7 @@ def ensure_rgb(image):
         return image.convert('RGB')
     return image
 
+@spaces.GPU
 def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
     if use_dotenv:
         load_dotenv()