mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-07-07 20:39:48 +00:00
[Inference] resolve rebase conflicts
fix
This commit is contained in:
parent
791a9fbb48
commit
dec8fdb229
@ -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
|
||||
|
@ -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.
|
||||
|
Loading…
Reference in New Issue
Block a user