lint(api): return models in order
This commit is contained in:
parent
c0ca7cf62f
commit
33eebacc0c
|
@ -218,15 +218,15 @@ def load_models(context: ServerContext):
|
||||||
path.join(context.model_path, 'diffusion-*'))]
|
path.join(context.model_path, 'diffusion-*'))]
|
||||||
diffusion_models.extend([
|
diffusion_models.extend([
|
||||||
get_model_name(f) for f in glob(path.join(context.model_path, 'stable-diffusion-*'))])
|
get_model_name(f) for f in glob(path.join(context.model_path, 'stable-diffusion-*'))])
|
||||||
diffusion_models = list(set(diffusion_models))
|
diffusion_models = list(set(diffusion_models)).sort()
|
||||||
|
|
||||||
correction_models = [
|
correction_models = [
|
||||||
get_model_name(f) for f in glob(path.join(context.model_path, 'correction-*'))]
|
get_model_name(f) for f in glob(path.join(context.model_path, 'correction-*'))]
|
||||||
correction_models = list(set(correction_models))
|
correction_models = list(set(correction_models)).sort()
|
||||||
|
|
||||||
upscaling_models = [
|
upscaling_models = [
|
||||||
get_model_name(f) for f in glob(path.join(context.model_path, 'upscaling-*'))]
|
get_model_name(f) for f in glob(path.join(context.model_path, 'upscaling-*'))]
|
||||||
upscaling_models = list(set(upscaling_models))
|
upscaling_models = list(set(upscaling_models)).sort()
|
||||||
|
|
||||||
|
|
||||||
def load_params(context: ServerContext):
|
def load_params(context: ServerContext):
|
||||||
|
|
Loading…
Reference in New Issue