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

pull/3313/head
Ziheng Qin 2023-03-28 11:36:11 -07:00 committed by binmakeswell
parent 964a28678f
commit 1bed38ef37
1 changed files with 1 additions and 0 deletions

View File

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