Update llama.py

This commit is contained in:
Daniel Han 2024-07-23 11:16:40 -07:00
commit 5da00a946e

View file

@ -1065,7 +1065,7 @@ class LlamaExtendedRotaryEmbedding(torch.nn.Module):
config = None, # [TODO] Hack to pass in config - need to remove later
):
super().__init__()
print(__FILE__, __LINE__)
print(__LINE__)
# if config is not None: return # [TODO] Hack to pass in config - need to remove later
self.dim = dim