mirror of https://github.com/hpcaitech/ColossalAI
dd2bf02679
* 1.support sam 2.add fused qkv for nn.Linear * update utils support set element in list * overtwrite SamVisionAttention foward to use DropoutForParallelInput * remove unused code |
||
---|---|---|
.. | ||
model_zoo | ||
__init__.py |