From 2566489b830d6c252557ed635ad8265b74c8fb6b Mon Sep 17 00:00:00 2001 From: artyomko Date: Tue, 10 Dec 2024 14:46:40 +0000 Subject: [PATCH] black --- sonar/__init__.py | 3 ++- sonar/models/mutox/loader.py | 4 +++- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/sonar/__init__.py b/sonar/__init__.py index a741a47..ecb3129 100644 --- a/sonar/__init__.py +++ b/sonar/__init__.py @@ -9,7 +9,7 @@ """ from fairseq2.assets import default_asset_store -__version__ = "0.2.1" +__version__ = "0.3.1" def setup_fairseq2() -> None: @@ -17,4 +17,5 @@ def setup_fairseq2() -> None: # the directory /cards. default_asset_store.add_package_metadata_provider("sonar.cards") + setup_fairseq2() diff --git a/sonar/models/mutox/loader.py b/sonar/models/mutox/loader.py index b6ce64a..a1137d1 100644 --- a/sonar/models/mutox/loader.py +++ b/sonar/models/mutox/loader.py @@ -32,7 +32,9 @@ def convert_mutox_checkpoint( return {"model": new_dict} -load_mutox_config = StandardModelConfigLoader(family="mutox", config_kls=MutoxConfig, arch_configs=mutox_archs) +load_mutox_config = StandardModelConfigLoader( + family="mutox", config_kls=MutoxConfig, arch_configs=mutox_archs +) load_mutox_model = StandardModelLoader( config_loader=load_mutox_config,