ds1david commited on
Commit
888435a
·
1 Parent(s): 357df1b

fixing bugs

Browse files
Files changed (1) hide show
  1. app.py +22 -23
app.py CHANGED
@@ -87,31 +87,30 @@ except Exception as e:
87
  pipe = None
88
  modelo_profundidade = None
89
 
90
- if device == "cuda":
91
- try:
92
- logger.divider("Configurando Pipeline de Arte")
93
- pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
94
- "stabilityai/stable-diffusion-xl-base-1.0",
95
- torch_dtype=torch_dtype,
96
- variant="fp16",
97
- use_safetensors=True
98
- ).to(device)
99
-
100
- pipe.load_lora_weights(
101
- "KappaNeuro/bas-relief",
102
- weight_name="BAS-RELIEF.safetensors"
103
- )
104
- logger.success("Pipeline SDXL configurado")
105
 
106
- logger.etapa("Configurando Modelo de Profundidade")
107
- processador_profundidade = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
108
- modelo_profundidade = DPTForDepthEstimation.from_pretrained("Intel/dpt-large").to(device)
109
- logger.success("Modelo de profundidade pronto")
110
 
111
- except Exception as e:
112
- logger.error(f"Erro na configuração da GPU: {str(e)}")
113
- pipe = None
114
- modelo_profundidade = None
115
 
116
 
117
  # ================== FLUXO DE PROCESSAMENTO ==================
 
87
  pipe = None
88
  modelo_profundidade = None
89
 
90
+ try:
91
+ logger.divider("Configurando Pipeline de Arte")
92
+ pipe = StableDiffusionXLImg2ImgPipeline.from_pretrained(
93
+ "stabilityai/stable-diffusion-xl-base-1.0",
94
+ torch_dtype=torch_dtype,
95
+ variant="fp16",
96
+ use_safetensors=True
97
+ ).to(device)
98
+
99
+ pipe.load_lora_weights(
100
+ "KappaNeuro/bas-relief",
101
+ weight_name="BAS-RELIEF.safetensors"
102
+ )
103
+ logger.success("Pipeline SDXL configurado")
 
104
 
105
+ logger.etapa("Configurando Modelo de Profundidade")
106
+ processador_profundidade = DPTFeatureExtractor.from_pretrained("Intel/dpt-large")
107
+ modelo_profundidade = DPTForDepthEstimation.from_pretrained("Intel/dpt-large").to(device)
108
+ logger.success("Modelo de profundidade pronto")
109
 
110
+ except Exception as e:
111
+ logger.error(f"Erro na configuração da GPU: {str(e)}")
112
+ pipe = None
113
+ modelo_profundidade = None
114
 
115
 
116
  # ================== FLUXO DE PROCESSAMENTO ==================