hf/bench.py: need to specify bfloat16 otherwise it consumes twice as much memory in A10.#9
Open
ivanbaldo wants to merge 1 commit intohamelsmu:masterfrom
Open
hf/bench.py: need to specify bfloat16 otherwise it consumes twice as much memory in A10.#9ivanbaldo wants to merge 1 commit intohamelsmu:masterfrom
ivanbaldo wants to merge 1 commit intohamelsmu:masterfrom