Update llama.py
This commit is contained in:
parent
1cc70ed7fa
commit
55e0ec3e26
1 changed files with 1 additions and 1 deletions
|
|
@ -1073,7 +1073,7 @@ class LlamaExtendedRotaryEmbedding(torch.nn.Module):
|
|||
device = "cuda"
|
||||
max_position_embeddings = config.max_position_embeddings
|
||||
pass
|
||||
|
||||
|
||||
self.dim = dim
|
||||
self.max_position_embeddings = max_position_embeddings
|
||||
self.base = base
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue