mirror of
https://github.com/leon-ai/leon.git
synced 2024-08-16 21:50:33 +03:00
refactor(scripts): differentiate PyTorch info log on macOS
This commit is contained in:
parent
6c3abf30b0
commit
8dc725a697
@ -147,7 +147,12 @@ SPACY_MODELS.set('fr', {
|
||||
* @see https://docs.nvidia.com/deeplearning/cudnn/latest/reference/support-matrix.html
|
||||
*/
|
||||
const installPytorch = async () => {
|
||||
LogHelper.info('Installing PyTorch with CUDA support...')
|
||||
const logInfo =
|
||||
osType === OSTypes.MacOS
|
||||
? 'Installing PyTorch...'
|
||||
: 'Installing PyTorch with CUDA support...'
|
||||
LogHelper.info(logInfo)
|
||||
|
||||
try {
|
||||
// There is no CUDA support on macOS
|
||||
const commandToExecute =
|
||||
|
Loading…
Reference in New Issue
Block a user