JOHANhuynhhhhh commited on
Commit
c43c3f3
·
verified ·
1 Parent(s): 84e81bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -70,7 +70,7 @@ inpainting_models = OrderedDict([
70
  ("Stable-Inpainting 2.0", 'sd2_inp'),
71
  ("Stable-Inpainting 1.5", 'sd15_inp')
72
  ])
73
- sr_model = models.sd2_sr.load_model(device='cuda:1')
74
  sam_predictor = models.sam.load_model(device='cuda:0')
75
 
76
  inp_model_name = list(inpainting_models.keys())[0]
 
70
  ("Stable-Inpainting 2.0", 'sd2_inp'),
71
  ("Stable-Inpainting 1.5", 'sd15_inp')
72
  ])
73
+ sr_model = models.sd2_sr.load_model(device='cuda:0')
74
  sam_predictor = models.sam.load_model(device='cuda:0')
75
 
76
  inp_model_name = list(inpainting_models.keys())[0]