Ufoptg commited on
Commit
6fc27bf
·
verified ·
1 Parent(s): b1fb56a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +8 -1
main.py CHANGED
@@ -259,7 +259,14 @@ def dalle_3xl(
259
  ):
260
  API_URL = SOURCE_DALLE3XL_URL
261
  try:
262
- payload = {"inputs": item.query}
 
 
 
 
 
 
 
263
  headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
264
  response = requests.post(API_URL, headers=headers, json=payload)
265
  response.raise_for_status()
 
259
  ):
260
  API_URL = SOURCE_DALLE3XL_URL
261
  try:
262
+ payload = {
263
+ "inputs": item.query,
264
+ "is_negative": item.is_neg,
265
+ "steps": item.steps,
266
+ "cfg_scale": item.cfg_scale,
267
+ "seed": item.seed if item.seed != -1 else random.randint(1, 1000000000),
268
+ "strength": item.strength
269
+ }
270
  headers = {"Authorization": f"Bearer {HUGGING_TOKEN}"}
271
  response = requests.post(API_URL, headers=headers, json=payload)
272
  response.raise_for_status()