@spaces.GPU

scripts/anime.py

@@ -28,6 +28,7 @@ def init_model(use_local=False):
     model.eval()
 
 # numpy配列の画像を受け取り、線画を生成してnumpy配列で返す
+@spaces.GPU
 def generate_sketch(image, clahe_clip=-1, load_size=512):
     """
     Generate sketch image from input image

scripts/process_utils.py

@@ -51,6 +51,7 @@ def initialize(_use_local=False, use_gpu=False, use_dotenv=False):
     sotai_gen_pipe = initialize_sotai_model()
     refine_gen_pipe = initialize_refine_model()
 
+@spaces.GPU
 def load_lora(pipeline, lora_path, alpha=0.75):
     pipeline.load_lora_weights(lora_path)
     pipeline.fuse_lora(lora_scale=alpha)