ds1david commited on
Commit
1f384c6
·
1 Parent(s): f41a4a7

fixing bugs

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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 = "cuda" if torch.cuda.is_available() else "cpu"
33
- torch_dtype = torch.float16 if device == "cuda" else torch.float32
 
 
 
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",