Spaces:
Paused
Paused
Update app.py
Browse files
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:
|
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]
|