"vscode:/vscode.git/clone" did not exist on "e513c16e82221e0a150b97a31f2c3c864913c454"
Unverified Commit e5f7e03b authored by Sanchit Gandhi's avatar Sanchit Gandhi Committed by GitHub
Browse files

[Permisson] Style fix (#26228)

fix copies
parent e4e55af7
......@@ -72,7 +72,7 @@ def _expand_mask(mask: torch.Tensor, dtype: torch.dtype, tgt_len: Optional[int]
# Copied from transformers.models.llama.modeling_llama.LlamaRotaryEmbedding with Llama->Persimmon
class PersimmonRotaryEmbedding(torch.nn.Module):
class PersimmonRotaryEmbedding(nn.Module):
def __init__(self, dim, max_position_embeddings=2048, base=10000, device=None):
super().__init__()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment