Update app.py

app.py

@@ -43,8 +43,15 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
 
 pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
-# 텍스트 인코더의 dtype 불일치를 막기 위해 float16으로 변환
+# 텍스트 인코더를 float16으로 강제 변환
 pipe.text_encoder = pipe.text_encoder.to("cuda", dtype=torch.float16)
+# 추가: text_projection의 forward를 오버라이딩하여 입력이 float16이 아니면 half로 캐스팅
+original_text_projection_forward = pipe.text_encoder.text_projection.forward
+def fixed_text_projection_forward(x):
+    if x.dtype != torch.float16:
+        x = x.half()
+    return original_text_projection_forward(x)
+pipe.text_encoder.text_projection.forward = fixed_text_projection_forward
 
 def can_expand(source_width, source_height, target_width, target_height, alignment):
     """Checks if the image can be expanded based on the alignment."""