pin Hindsight retrieval models to GPU0
This commit is contained in:
@@ -11,6 +11,7 @@ services:
|
||||
- float16
|
||||
environment:
|
||||
HF_HUB_CACHE: /data
|
||||
NVIDIA_VISIBLE_DEVICES: "0"
|
||||
HTTP_PROXY: http://192.168.50.45:7890
|
||||
HTTPS_PROXY: http://192.168.50.45:7890
|
||||
NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24
|
||||
@@ -49,6 +50,7 @@ services:
|
||||
- float16
|
||||
environment:
|
||||
HF_HUB_CACHE: /data
|
||||
NVIDIA_VISIBLE_DEVICES: "0"
|
||||
HTTP_PROXY: http://192.168.50.45:7890
|
||||
HTTPS_PROXY: http://192.168.50.45:7890
|
||||
NO_PROXY: 127.0.0.1,localhost,192.168.50.0/24
|
||||
|
||||
Reference in New Issue
Block a user