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

@@ -1,7 +1,7 @@
# Adapted from llama2.py
# Modify details for the adaptation of Qwen2 model.
"""Inference-only Qwen2 model compatible with HuggingFace weights."""
from typing import Any, Dict, Optional, Tuple
from typing import Any, Dict, List, Optional, Tuple
import torch
from torch import nn