test

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import torch
 import numpy as np
+import spaces
 from huggingface_hub import hf_hub_download
 from utils import utils, tools, preprocess
 
@@ -12,6 +13,7 @@ CONTROLNET_FILENAME = "ControlAny-SDXL/anime_canny/controlnet.safetensors"
 CACHE_DIR = None
 
 
+@spaces.GPU
 def ui():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model_file = hf_hub_download(
@@ -39,7 +41,7 @@ def ui():
         device=device,
         hf_cache_dir=CACHE_DIR,
         use_safetensors=True,
-        enable_xformers_memory_efficient_attention=torch.cuda.is_available(),
+        enable_xformers_memory_efficient_attention=True,
     )
 
     preprocessors = ['canny']