yangtb24 commited on
Commit
78e3cfc
·
verified ·
1 Parent(s): 9bef950

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -3
app.py CHANGED
@@ -15,7 +15,7 @@ from apscheduler.schedulers.background import BackgroundScheduler
15
  from flask import Flask, request, jsonify, Response, stream_with_context
16
 
17
  os.environ['TZ'] = 'Asia/Shanghai'
18
- time.tzset()
19
 
20
  logging.basicConfig(level=logging.INFO,
21
  format='%(asctime)s - %(levelname)s - %(message)s')
@@ -858,6 +858,9 @@ def handsome_images_generations():
858
  siliconflow_data["safety_tolerance"] = data.get("safety_tolerance", 2)
859
  siliconflow_data["interval"] = data.get("interval", 2)
860
  siliconflow_data["output_format"] = data.get("output_format", "png")
 
 
 
861
 
862
  if siliconflow_data["width"] < 256 or siliconflow_data["width"] > 1440 or siliconflow_data["width"] % 32 != 0:
863
  siliconflow_data["width"] = 1024
@@ -898,7 +901,7 @@ def handsome_images_generations():
898
  siliconflow_data["guidance_scale"] = 0
899
  if siliconflow_data["guidance_scale"] > 100:
900
  siliconflow_data["guidance_scale"] = 100
901
- # Validate image_size
902
  if "image_size" in siliconflow_data and siliconflow_data["image_size"] not in ["1024x1024", "512x1024", "768x512", "768x1024", "1024x576", "576x1024","960x1280", "720x1440", "720x1280"]:
903
  siliconflow_data["image_size"] = "1024x1024"
904
 
@@ -1047,7 +1050,9 @@ def handsome_chat_completions():
1047
  siliconflow_data["safety_tolerance"] = data.get("safety_tolerance", 2)
1048
  siliconflow_data["interval"] = data.get("interval", 2)
1049
  siliconflow_data["output_format"] = data.get("output_format", "png")
1050
-
 
 
1051
  if siliconflow_data["width"] < 256 or siliconflow_data["width"] > 1440 or siliconflow_data["width"] % 32 != 0:
1052
  siliconflow_data["width"] = 1024
1053
  if siliconflow_data["height"] < 256 or siliconflow_data["height"] > 1440 or siliconflow_data["height"] % 32 != 0:
 
15
  from flask import Flask, request, jsonify, Response, stream_with_context
16
 
17
  os.environ['TZ'] = 'Asia/Shanghai'
18
+ # time.tzset()
19
 
20
  logging.basicConfig(level=logging.INFO,
21
  format='%(asctime)s - %(levelname)s - %(message)s')
 
858
  siliconflow_data["safety_tolerance"] = data.get("safety_tolerance", 2)
859
  siliconflow_data["interval"] = data.get("interval", 2)
860
  siliconflow_data["output_format"] = data.get("output_format", "png")
861
+ seed = data.get("seed")
862
+ if isinstance(seed, int) and 0 < seed < 9999999999:
863
+ siliconflow_data["seed"] = seed
864
 
865
  if siliconflow_data["width"] < 256 or siliconflow_data["width"] > 1440 or siliconflow_data["width"] % 32 != 0:
866
  siliconflow_data["width"] = 1024
 
901
  siliconflow_data["guidance_scale"] = 0
902
  if siliconflow_data["guidance_scale"] > 100:
903
  siliconflow_data["guidance_scale"] = 100
904
+
905
  if "image_size" in siliconflow_data and siliconflow_data["image_size"] not in ["1024x1024", "512x1024", "768x512", "768x1024", "1024x576", "576x1024","960x1280", "720x1440", "720x1280"]:
906
  siliconflow_data["image_size"] = "1024x1024"
907
 
 
1050
  siliconflow_data["safety_tolerance"] = data.get("safety_tolerance", 2)
1051
  siliconflow_data["interval"] = data.get("interval", 2)
1052
  siliconflow_data["output_format"] = data.get("output_format", "png")
1053
+ seed = data.get("seed")
1054
+ if isinstance(seed, int) and 0 < seed < 9999999999:
1055
+ siliconflow_data["seed"] = seed
1056
  if siliconflow_data["width"] < 256 or siliconflow_data["width"] > 1440 or siliconflow_data["width"] % 32 != 0:
1057
  siliconflow_data["width"] = 1024
1058
  if siliconflow_data["height"] < 256 or siliconflow_data["height"] > 1440 or siliconflow_data["height"] % 32 != 0: