fix(api): dedupe models after removing extension
This commit is contained in:
parent
7cd0345cd2
commit
c0ca7cf62f
|
@ -218,11 +218,15 @@ def load_models(context: ServerContext):
|
|||
path.join(context.model_path, 'diffusion-*'))]
|
||||
diffusion_models.extend([
|
||||
get_model_name(f) for f in glob(path.join(context.model_path, 'stable-diffusion-*'))])
|
||||
diffusion_models = list(set(diffusion_models))
|
||||
|
||||
correction_models = [
|
||||
get_model_name(f) for f in glob(path.join(context.model_path, 'correction-*'))]
|
||||
correction_models = list(set(correction_models))
|
||||
|
||||
upscaling_models = [
|
||||
get_model_name(f) for f in glob(path.join(context.model_path, 'upscaling-*'))]
|
||||
upscaling_models = list(set(upscaling_models))
|
||||
|
||||
|
||||
def load_params(context: ServerContext):
|
||||
|
|
Loading…
Reference in New Issue