Update modules/app.py

modules/app.py

@@ -165,20 +165,20 @@ async def img2location(request: Request):
 @app.post("/pixart-sigma", dependencies=[Depends(verify_user_agent)])
 async def pixart_sigma(request: Request):
     request_json = await request.json()
-    prompt = str(request_json.get("prompt")) if request_json.get("prompt") is not None else None
-    negative_prompt = str(request_json.get("negative_prompt")) if request_json.get("negative_prompt") is not None else ""
-    style = str(request_json.get("style")) if request_json.get("style") is not None else "(No style)"
-    use_negative_prompt = bool(request_json.get("use_negative_prompt")) if request_json.get("use_negative_prompt") is not None else True
-    num_imgs = float(request_json.get("num_imgs")) if request_json.get("num_imgs") is not None else 1
-    seed = float(request_json.get("seed")) if request_json.get("seed") is not None else 0
-    width = float(request_json.get("width")) if request_json.get("width") is not None else 1024
-    height = float(request_json.get("height")) if request_json.get("height") is not None else 1024
-    schedule = str(request_json.get("schedule")) if request_json.get("schedule") is not None else "DPM-Solver"
-    dpms_guidance_scale = float(request_json.get("dpms_guidance_scale")) if request_json.get("dpms_guidance_scale") is not None else 4.5
-    sas_guidance_scale = float(request_json.get("sas_guidance_scale")) if request_json.get("sas_guidance_scale") is not None else 3
-    dpms_inference_steps = float(request_json.get("dpms_inference_steps")) if request_json.get("dpms_inference_steps") is not None else 14
-    sas_inference_steps = float(request_json.get("sas_inference_steps")) if request_json.get("sas_inference_steps") is not None else 25
-    randomize_seed = bool(request_json.get("randomize_seed")) if request_json.get("randomize_seed") is not None else True
+    prompt = request_json.get("prompt", None)
+    negative_prompt = request_json.get("negative_prompt", "")
+    style = request_json.get("style", "(No style)")
+    use_negative_prompt = request_json.get("use_negative_prompt", True)
+    num_imgs = request_json.get("num_imgs", 1)
+    seed = request_json.get("seed", 0)
+    width = request_json.get("width", 1024)
+    height = request_json.get("height", 1024)
+    schedule = request_json.get("schedule", "DPM-Solver")
+    dpms_guidance_scale = request_json.get("dpms_guidance_scale", 4.5)
+    sas_guidance_scale = request_json.get("sas_guidance_scale", 3)
+    dpms_inference_steps = request_json.get("dpms_inference_steps", 14)
+    sas_inference_steps = request_json.get("sas_inference_steps", 25)
+    randomize_seed = request_json.get("randomize_seed", True)
 
     hash = generate_hash()
 
@@ -205,7 +205,7 @@ async def pixart_sigma(request: Request):
             sas_guidance_scale,
             dpms_inference_steps,
             sas_inference_steps,
-            randomize_seed,
+            True,
         ],
         'event_data': None,
         'fn_index': 3,