Skip to content

Commit

Permalink
🤖 Lint code
Browse files Browse the repository at this point in the history
  • Loading branch information
ivy-branch committed Aug 6, 2023
1 parent 3fd2039 commit 6bea0fa
Showing 1 changed file with 27 additions and 27 deletions.
54 changes: 27 additions & 27 deletions ivy_models/bart/config_bart.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,33 +216,33 @@ class BartConfig(BaseSpec):
"num_attention_heads": "encoder_attention_heads",
"hidden_size": "d_model",
}
vocab_size: int = 50265,
max_position_embeddings: int = 1024,
encoder_layers: int = 12,
encoder_ffn_dim: int = 4096,
encoder_attention_heads: int = 16,
decoder_layers: int = 12,
decoder_ffn_dim: int = 4096,
decoder_attention_heads: int = 16,
encoder_layerdrop: float = 0.0,
decoder_layerdrop: float = 0.0,
activation_function: str = "gelu",
d_model: int = 1024,
dropout: float = 0.1,
attention_dropout: float = 0.0,
activation_dropout: float = 0.0,
init_std: float = 0.02,
classifier_dropout: float = 0.0,
scale_embedding: bool = False,
use_cache: bool = True,
num_labels: int = 3,
pad_token_id: int = 1,
bos_token_id: int = 0,
eos_token_id: int = 2,
is_encoder_decoder: bool = True,
decoder_start_token_id: int = 2,
forced_eos_token_id: int = 2,

vocab_size: int = (50265,)
max_position_embeddings: int = (1024,)
encoder_layers: int = (12,)
encoder_ffn_dim: int = (4096,)
encoder_attention_heads: int = (16,)
decoder_layers: int = (12,)
decoder_ffn_dim: int = (4096,)
decoder_attention_heads: int = (16,)
encoder_layerdrop: float = (0.0,)
decoder_layerdrop: float = (0.0,)
activation_function: str = ("gelu",)
d_model: int = (1024,)
dropout: float = (0.1,)
attention_dropout: float = (0.0,)
activation_dropout: float = (0.0,)
init_std: float = (0.02,)
classifier_dropout: float = (0.0,)
scale_embedding: bool = (False,)
use_cache: bool = (True,)
num_labels: int = (3,)
pad_token_id: int = (1,)
bos_token_id: int = (0,)
eos_token_id: int = (2,)
is_encoder_decoder: bool = (True,)
decoder_start_token_id: int = (2,)
forced_eos_token_id: int = (2,)

def get(self, *attr_names):
new_dict = {}
Expand Down

0 comments on commit 6bea0fa

Please sign in to comment.