Skip to content

Commit

Permalink
small fix warning
Browse files Browse the repository at this point in the history
  • Loading branch information
albertz committed Jan 13, 2025
1 parent 2332837 commit 0428e25
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions returnn/frontend/conversions/hf_llama.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@
from returnn.frontend.decoder.transformer import TransformerDecoder, TransformerDecoderLayer, FeedForwardGated

if TYPE_CHECKING:
# noinspection PyUnresolvedReferences,PyPackageRequirements,PyProtectedMemberInspection
# noinspection PyUnresolvedReferences,PyPackageRequirements,PyProtectedMember
from transformers.models.llama.modeling_llama import (
LlamaModel,
LlamaForCausalLM,
Expand All @@ -27,7 +27,7 @@ def import_params_hf_llama_to_rf_transformer_decoder(
"""
import torch

# noinspection PyUnresolvedReferences,PyPackageRequirements,PyProtectedMemberInspection
# noinspection PyUnresolvedReferences,PyPackageRequirements,PyProtectedMember
from transformers.models.llama.modeling_llama import LlamaModel, LlamaForCausalLM, LlamaDecoderLayer

print("HF Model:")
Expand Down

0 comments on commit 0428e25

Please sign in to comment.