Hatman commited on
Commit
64b1b83
·
verified ·
1 Parent(s): c010994

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -262,6 +262,7 @@ def get_random_model(models):
262
  global last_two_models
263
  model = None
264
  priorities = [
 
265
  "kandinsky-community",
266
  "Kolors-diffusers",
267
  "Juggernaut",
@@ -282,7 +283,7 @@ def get_random_model(models):
282
  print("Choosing randomly")
283
  model = random.choice(models)
284
  last_two_models.append(model)
285
- last_two_models = last_two_models[-5:]
286
  return model
287
 
288
  def nsfw_check(attempts = 1):
 
262
  global last_two_models
263
  model = None
264
  priorities = [
265
+ "FLUX",
266
  "kandinsky-community",
267
  "Kolors-diffusers",
268
  "Juggernaut",
 
283
  print("Choosing randomly")
284
  model = random.choice(models)
285
  last_two_models.append(model)
286
+ last_two_models = last_two_models[-3:]
287
  return model
288
 
289
  def nsfw_check(attempts = 1):