riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-16 11:08:23 +06:00
Code Issues Actions 28 Packages Projects Releases Wiki Activity

Fix quality

Browse Source
This commit is contained in:
Sylvain Gugger 2023-02-22 03:36:15 -05:00
parent 24b930ad1d
commit ee6e71e29c
No known key found for this signature in database
GPG Key ID: 4198F83D867EFC2E
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pyproject.toml
View File

@ -4,7 +4,7 @@ target-version = ['py37']
[tool.ruff] [tool.ruff]
# Never enforce `E501` (line length violations). # Never enforce `E501` (line length violations).
ignore = ["E501", "E741", "W605"] ignore = ["C901", "E501", "E741", "W605"]
select = ["C", "E", "F", "I", "W"] select = ["C", "E", "F", "I", "W"]
line-length = 119 line-length = 119

4
src/transformers/models/efficientnet/convert_efficientnet_to_pytorch.py
View File

@ -160,7 +160,7 @@ def convert_image_processor(model_name):
# here we list all keys to be renamed (original name on the left, our name on the right) # here we list all keys to be renamed (original name on the left, our name on the right)
def rename_keys(original_param_names): def rename_keys(original_param_names):
block_names = [v.split("_")[0].split("block")[1] for v in original_param_names if v.startswith("block")] block_names = [v.split("_")[0].split("block")[1] for v in original_param_names if v.startswith("block")]
block_names = sorted(list(set(block_names))) block_names = sorted(set(block_names))
num_blocks = len(block_names) num_blocks = len(block_names)
block_name_mapping = {b: str(i) for b, i in zip(block_names, range(num_blocks))} block_name_mapping = {b: str(i) for b, i in zip(block_names, range(num_blocks))}
@ -267,7 +267,7 @@ def convert_efficientnet_checkpoint(model_name, pytorch_dump_folder_path, save_m
tf_params = {param.name: param.numpy() for param in tf_params} tf_params = {param.name: param.numpy() for param in tf_params}
for param in tf_non_train_params: for param in tf_non_train_params:
tf_params[param.name] = param.numpy() tf_params[param.name] = param.numpy()
tf_param_names = [k for k in tf_params.keys()] tf_param_names = list(tf_params.keys())
# Load HuggingFace model # Load HuggingFace model
config = get_efficientnet_config(model_name) config = get_efficientnet_config(model_name)

2
src/transformers/models/gptsan_japanese/modeling_gptsan_japanese.py
View File

@ -1029,7 +1029,7 @@ class GPTSanJapaneseModel(GPTSanJapanesePreTrainedModel):
) # n_layer x batch x n_heads x N x N ) # n_layer x batch x n_heads x N x N
# outputs # outputs
present_key_value_states = tuple() if self.config.use_cache or use_cache else None present_key_value_states = () if self.config.use_cache or use_cache else None
all_hidden_states = () if self.config.output_hidden_states or output_hidden_states else None all_hidden_states = () if self.config.output_hidden_states or output_hidden_states else None
all_attentions = () if self.config.output_attentions or output_attentions else None all_attentions = () if self.config.output_attentions or output_attentions else None
all_router_probs = () if self.config.output_router_logits or output_router_logits else None all_router_probs = () if self.config.output_router_logits or output_router_logits else None