[NFC] polish colossalai/cli/benchmark/models.py code style (#3290)

This commit is contained in:
Ziheng Qin
2023-03-28 11:36:11 -07:00
committed by binmakeswell
parent 964a28678f
commit 1bed38ef37

View File

@@ -1,4 +1,5 @@
import torch import torch
import colossalai.nn as col_nn import colossalai.nn as col_nn