test

app.py

@@ -13,13 +13,6 @@ CONTROLNET_FILENAME = "ControlAny-SDXL/anime_canny/controlnet.safetensors"
 CACHE_DIR = None
 
 
-@spaces.GPU
-def optimize_pipeline(pipeline):
-    pipeline.to(dtype=torch.float16)
-    # pipeline.enable_xformers_memory_efficient_attention()
-    return pipeline
-
-
 def ui():
     device = "cuda" if torch.cuda.is_available() else "cpu"
     model_file = hf_hub_download(
@@ -48,9 +41,7 @@ def ui():
         device=device,
         hf_cache_dir=CACHE_DIR,
         use_safetensors=True,
-        enable_xformers_memory_efficient_attention=torch.cuda.is_available(),
     )
-    pipeline = optimize_pipeline(pipeline)
 
     preprocessors = ['canny']
     schedulers = ['Euler A', 'UniPC', 'Euler', 'DDIM', 'DDPM']