Update app.py

app.py

@@ -45,13 +45,14 @@ pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
 
 # 텍스트 인코더를 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
+# 만약 text_projection 속성이 있다면, 입력이 float16이 아니면 half로 캐스팅하도록 오버라이딩
+if hasattr(pipe.text_encoder, "text_projection"):
+    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."""