Spaces:
Paused
Paused
Commit
·
ce75bc0
1
Parent(s):
b2e5e8f
Update app_v3.py
Browse files
app_v3.py
CHANGED
@@ -7,7 +7,8 @@ import torch
|
|
7 |
import subprocess
|
8 |
|
9 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
|
10 |
-
os.environ[
|
|
|
11 |
|
12 |
|
13 |
# Define pretrained and quantized model directories
|
|
|
7 |
import subprocess
|
8 |
|
9 |
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = 'max_split_size_mb:128'
|
10 |
+
os.environ["CUDA_DEVICE_ORDER"]="PCI_BUS_ID"
|
11 |
+
os.environ["CUDA_VISIBLE_DEVICES"]="1"
|
12 |
|
13 |
|
14 |
# Define pretrained and quantized model directories
|