mirror of
https://github.com/pinokiofactory/flux-webui.git
synced 2024-10-05 23:57:57 +03:00
76 lines
2.8 KiB
JavaScript
76 lines
2.8 KiB
JavaScript
module.exports = {
|
|
run: [
|
|
// windows nvidia
|
|
{
|
|
"when": "{{platform === 'win32' && gpu === 'nvidia'}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch==2.4.0 torchvision torchaudio==2.4.0 {{args && args.xformers ? 'xformers' : ''}} --index-url https://download.pytorch.org/whl/cu121"
|
|
}
|
|
},
|
|
// windows amd
|
|
{
|
|
"when": "{{platform === 'win32' && gpu === 'amd'}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch-directml torchvision torchaudio"
|
|
}
|
|
},
|
|
// windows cpu
|
|
{
|
|
"when": "{{platform === 'win32' && (gpu !== 'nvidia' && gpu !== 'amd')}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch==2.4.0 torchvision torchaudio==2.4.0"
|
|
}
|
|
},
|
|
// mac
|
|
{
|
|
"when": "{{platform === 'darwin'}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch==2.3.1 torchvision torchaudio==2.3.1"
|
|
//"message": "pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cpu"
|
|
}
|
|
},
|
|
// linux nvidia
|
|
{
|
|
"when": "{{platform === 'linux' && gpu === 'nvidia'}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch==2.4.0 torchvision torchaudio==2.4.0 {{args && args.xformers ? 'xformers' : ''}} --index-url https://download.pytorch.org/whl/cu121"
|
|
}
|
|
},
|
|
// linux rocm (amd)
|
|
{
|
|
"when": "{{platform === 'linux' && gpu === 'amd'}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch==2.4.0 torchvision torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/rocm6.0"
|
|
}
|
|
},
|
|
// linux cpu
|
|
{
|
|
"when": "{{platform === 'linux' && (gpu !== 'amd' && gpu !=='nvidia')}}",
|
|
"method": "shell.run",
|
|
"params": {
|
|
"venv": "{{args && args.venv ? args.venv : null}}",
|
|
"path": "{{args && args.path ? args.path : '.'}}",
|
|
"message": "pip install torch==2.4.0 torchvision torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cpu"
|
|
}
|
|
}
|
|
]
|
|
}
|