Update app.py

app.py

@@ -63,19 +63,25 @@ controlnet = FluxControlNetModel.from_pretrained(
     "jasperai/Flux.1-dev-Controlnet-Upscaler", torch_dtype=torch.bfloat16
 )
 
-# Upscale 파이프라인 설정 (기존 pipe 재사용)
-# Upscale 파이프라인 설정 (기존 pipe 재사용)
+print("Available attributes in FLUX pipeline:", pipe.__dict__.keys())
+
+# Upscale 파이프라인 설정 (FLUX 모델의 실제 구조에 맞게 수정)
 pipe_upscale = FluxControlNetPipeline(
     vae=pipe.vae,
     text_encoder=pipe.text_encoder,
     tokenizer=pipe.tokenizer,
-    transformer=pipe.transformer,  # FLUX 모델은 transformer를 사용할 수 있습니다
+    transformer=pipe.transformer,
     scheduler=pipe.scheduler,
-    safety_checker=pipe.safety_checker,
-    feature_extractor=pipe.feature_extractor,
     controlnet=controlnet
 )
 
+# 사용 가능한 경우에만 추가 속성 설정
+if hasattr(pipe, 'feature_extractor'):
+    pipe_upscale.feature_extractor = pipe.feature_extractor
+
+# 모든 파이프라인을 device로 이동
+pipe_upscale.to(device)
+
 # 모든 파이프라인을 device로 이동
 pipe_i2i.to(device)
 controlnet.to(device)