Revert removing the unused imports (#385)

This commit is contained in:
Liangsheng Yin
2024-04-23 22:36:33 +08:00
committed by GitHub
parent 9acc6e3504
commit 150d7020ed
33 changed files with 72 additions and 26 deletions

View File

@@ -20,7 +20,7 @@
# This file is based on the LLama model definition file in transformers
"""PyTorch Cohere model."""
from typing import Optional, Tuple
from typing import List, Optional, Tuple
import torch
import torch.utils.checkpoint