up

app.py

@@ -1,3 +1,6 @@
+import os
+os.environ['CUDA_DEVICE_ORDER'] = 'PCI_BUS_ID'
+os.environ['CUDA_VISIBLE_DEVICES'] = '2'
 import gradio as gr
 import numpy as np
 from models import make_inpainting
@@ -78,9 +81,6 @@ def upscale(image, prompt):
     # pipeline = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler", torch_dtype=torch.float16)
     # pipeline = pipeline.to("cuda")
 
-    #up
-
-    #pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
     pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
 
     ret = pipeline(prompt=prompt,