[Inference] resolve rebase conflicts

fix
This commit is contained in:
CjhHa1 2024-04-11 10:12:31 +08:00
parent 791a9fbb48
commit dec8fdb229
2 changed files with 1 additions and 2 deletions

View File

@ -1,6 +1,6 @@
import time
from itertools import count
from typing import Dict, List, Optional, Tuple, Union, Iterable
from typing import Dict, List, Optional, Tuple, Union
import numpy as np
import torch

View File

@ -175,7 +175,6 @@ class VocabParallelEmbedding1D(ParallelModule):
he initializer of weight, defaults to normal initializer.
The ``args`` and ``kwargs`` used in :class:``torch.nn.functional.embedding`` should contain:
::
max_norm (float, optional): If given, each embedding vector with norm larger than max_norm is
renormalized to have norm max_norm. Note: this will modify weight in-place.