test

app.py

@@ -31,7 +31,7 @@ def ui():
         cache_dir=CACHE_DIR,
     )
 
-    pipeline = spaces.GPU(tools.get_pipeline)(
+    pipeline = tools.get_pipeline(
         pretrained_model_name_or_path=model_file,
         unet_model_name_or_path=unet_file,
         controlnet_model_name_or_path=controlnet_file,
@@ -43,6 +43,8 @@ def ui():
         use_safetensors=True,
         enable_xformers_memory_efficient_attention=True,
     )
+    with spaces.GPU:
+        pipeline = pipeline.to(dtype=torch.float16)
 
     preprocessors = ['canny']
     schedulers = ['Euler A', 'UniPC', 'Euler', 'DDIM', 'DDPM']