set fp32 to avoid errors

model.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
 import gc
-import spaces
 
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 from controlnet_aux.util import HWC3
 from diffusers import (
@@ -53,9 +53,9 @@ class Model:
         ):
             return self.pipe
         model_id = CONTROLNET_MODEL_IDS[task_name]
-        controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
+        controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32)
         pipe = StableDiffusionControlNetPipeline.from_pretrained(
-            base_model_id, safety_checker=None, controlnet=controlnet, torch_dtype=torch.float16
+            base_model_id, safety_checker=None, controlnet=controlnet, torch_dtype=torch.float32
         )
         pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
         if self.device.type == "cuda":
@@ -87,7 +87,7 @@ class Model:
         torch.cuda.empty_cache()
         gc.collect()
         model_id = CONTROLNET_MODEL_IDS[task_name]
-        controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float16)
+        controlnet = ControlNetModel.from_pretrained(model_id, torch_dtype=torch.float32)
         controlnet.to(self.device)
         torch.cuda.empty_cache()
         gc.collect()

requirements.txt

@@ -8,3 +8,4 @@ kornia
 huggingface_hub
 gradio==4.26.0
 xformers
+mediapipe