fixing bugs
Browse files
app.py
CHANGED
@@ -29,8 +29,11 @@ model_edsr, params_edsr = load_thera_model(REPO_ID_EDSR)
|
|
29 |
model_rdn, params_rdn = load_thera_model(REPO_ID_RDN)
|
30 |
|
31 |
# ========== Configuração do SDXL + Depth ==========
|
32 |
-
device = "
|
33 |
-
torch_dtype = torch.
|
|
|
|
|
|
|
34 |
|
35 |
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
36 |
"stabilityai/stable-diffusion-xl-base-1.0",
|
|
|
29 |
model_rdn, params_rdn = load_thera_model(REPO_ID_RDN)
|
30 |
|
31 |
# ========== Configuração do SDXL + Depth ==========
|
32 |
+
device = "cpu"
|
33 |
+
torch_dtype = torch.float32
|
34 |
+
# device = "cuda" if torch.cuda.is_available() else "cpu"
|
35 |
+
# torch_dtype = torch.float16 if device == "cuda" else torch.float32
|
36 |
+
|
37 |
|
38 |
pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
|
39 |
"stabilityai/stable-diffusion-xl-base-1.0",
|