fix: Check device type in _initialize_llm
before moving to cuda.
#11485
Job | Run time |
---|---|
16m 48s | |
27m 57s | |
4s | |
23m 34s | |
14m 26s | |
23m 24s | |
20m 45s | |
19m 32s | |
7m 22s | |
15m 20s | |
7m 43s | |
26m 19s | |
29m 6s | |
11m 50s | |
28m 21s | |
15m 17s | |
4h 47m 48s |