fix upstream bug

swiss-ai · Jul 29, 2024 · 793bdf3 · 793bdf3
1 parent 590acd4
commit 793bdf3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/nanotron/models/llama.py b/src/nanotron/models/llama.py
@@ -14,7 +14,7 @@
 # limitations under the License.
 """PyTorch LLaMa model."""
 
-from typing import Dict, Optional, Union
+from typing import Dict, Optional, Union, List
 
 import torch
 from torch import nn