Minor changes

This commit is contained in:
oobabooga 2025-06-11 07:55:35 -07:00
parent 004fd8316c
commit 9d6a7f1bcf

View file

@ -236,14 +236,14 @@ if __name__ == "__main__":
settings_file = Path(shared.args.settings)
elif Path('user_data/settings.yaml').exists():
settings_file = Path('user_data/settings.yaml')
elif Path('user_data/settings.json').exists():
settings_file = Path('user_data/settings.json')
if settings_file is not None:
logger.info(f"Loading settings from \"{settings_file}\"")
file_contents = open(settings_file, 'r', encoding='utf-8').read()
new_settings = json.loads(file_contents) if settings_file.suffix == "json" else yaml.safe_load(file_contents)
shared.settings.update(new_settings)
with open(settings_file, 'r', encoding='utf-8') as f:
new_settings = yaml.safe_load(f.read())
if new_settings:
shared.settings.update(new_settings)
# Fallback settings for models
shared.model_config['.*'] = get_fallback_settings()