charliebaby2023 commited on
Commit
d0edf1a
·
verified ·
1 Parent(s): 7d3c435

Update handle_models.py

Browse files
Files changed (1) hide show
  1. handle_models.py +2 -0
handle_models.py CHANGED
@@ -1,4 +1,6 @@
1
  from config import howManyModelsToUse,num_models,max_images,inference_timeout,MAX_SEED,thePrompt,preSetPrompt,negPreSetPrompt
 
 
2
 
3
  def load_fn(models):
4
  global models_load
 
1
  from config import howManyModelsToUse,num_models,max_images,inference_timeout,MAX_SEED,thePrompt,preSetPrompt,negPreSetPrompt
2
+ from all_models import models
3
+ default_models = models[:num_models]
4
 
5
  def load_fn(models):
6
  global models_load