Add PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True env
This commit is contained in:
+1
-1
@@ -5,7 +5,7 @@ services:
|
|||||||
network_mode: bridge
|
network_mode: bridge
|
||||||
environment:
|
environment:
|
||||||
- HF_TOKEN=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
|
- HF_TOKEN=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
|
||||||
- PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
|
- PYTORCH_ALLOC_CONF=expandable_segments:True,throw_on_cudamalloc_oom:True
|
||||||
ipc: host
|
ipc: host
|
||||||
ports:
|
ports:
|
||||||
- "8000:8000"
|
- "8000:8000"
|
||||||
|
|||||||
Reference in New Issue
Block a user