Hatman commited on
Commit
25ec4e4
·
verified ·
1 Parent(s): bbc17d0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -262,7 +262,9 @@ def get_random_model(models):
262
  global last_two_models
263
  model = None
264
  priorities = [
265
- "FLUX",
 
 
266
  "kandinsky-community",
267
  "Kolors-diffusers",
268
  "Juggernaut",
@@ -283,7 +285,7 @@ def get_random_model(models):
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):
 
262
  global last_two_models
263
  model = None
264
  priorities = [
265
+ "stabilityai/stable-diffusion-3.5-large-turbo",
266
+ "stabilityai/stable-diffusion-3.5-large",
267
+ "black-forest-labs",
268
  "kandinsky-community",
269
  "Kolors-diffusers",
270
  "Juggernaut",
 
285
  print("Choosing randomly")
286
  model = random.choice(models)
287
  last_two_models.append(model)
288
+ last_two_models = last_two_models[-5:]
289
  return model
290
 
291
  def nsfw_check(attempts=1):