Update handler.py

handler.py

@@ -28,6 +28,8 @@ class EndpointHandler:
         parameters = cast(dict, data.get("parameters", {}))
         do_scaling = cast(bool, parameters.get("do_scaling", True))
         scaling_factor = cast(float, parameters.get("scaling_factor", None))
+        if scaling_factor is not None:
+            scaling_factor = float(scaling_factor)
         if do_scaling and scaling_factor is None:
             scaling_factor = self.vae.config.scaling_factor
         output_type = cast(str, parameters.get("output_type", "pil"))
@@ -58,4 +60,4 @@ class EndpointHandler:
         elif output_type == "pt":
             frames = frames
 
-        return frames
+        return frames