Update worker_runpod.py

worker_runpod.py

@@ -27,17 +27,6 @@ with torch.inference_mode():
     vae = VAELoader.load_vae("ae.safetensors")[0]
     controlnet = ControlNetLoader.load_controlnet("controlnet.safetensors")[0]
 
-def closestNumber(n, m):
-    q = int(n / m)
-    n1 = m * q
-    if (n * m) > 0:
-        n2 = m * (q + 1)
-    else:
-        n2 = m * (q - 1)
-    if abs(n - n1) < abs(n - n2):
-        return n1
-    return n2
-
 @torch.inference_mode()
 def generate(input):
     values = input["input"]