michaelapplydesign commited on
Commit
2361bd2
·
1 Parent(s): 6672974
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
3
- os.environ['CUDA_VISIBLE_DEVICES'] = '2'
4
  import gradio as gr
5
  import numpy as np
6
  from models import make_inpainting
 
1
  import os
2
  os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
3
+ os.environ['CUDA_VISIBLE_DEVICES'] = '0'
4
  import gradio as gr
5
  import numpy as np
6
  from models import make_inpainting