Skip to content

Commit

Permalink
black
Browse files Browse the repository at this point in the history
  • Loading branch information
artyomko committed Dec 10, 2024
1 parent 5df830c commit 2566489
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 2 deletions.
3 changes: 2 additions & 1 deletion sonar/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,12 +9,13 @@
"""
from fairseq2.assets import default_asset_store

__version__ = "0.2.1"
__version__ = "0.3.1"


def setup_fairseq2() -> None:
# Make sure that the default fairseq2 asset store can resolve cards under
# the directory <sonar>/cards.
default_asset_store.add_package_metadata_provider("sonar.cards")


setup_fairseq2()
4 changes: 3 additions & 1 deletion sonar/models/mutox/loader.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down

0 comments on commit 2566489

Please sign in to comment.