Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -43,6 +43,8 @@ pipe = StableDiffusionXLFillPipeline.from_pretrained(
|
|
43 |
|
44 |
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|
45 |
|
|
|
|
|
46 |
|
47 |
def can_expand(source_width, source_height, target_width, target_height, alignment):
|
48 |
"""Checks if the image can be expanded based on the alignment."""
|
|
|
43 |
|
44 |
pipe.scheduler = TCDScheduler.from_config(pipe.scheduler.config)
|
45 |
|
46 |
+
# 텍스트 인코더의 dtype 불일치를 막기 위해 float16으로 변환
|
47 |
+
pipe.text_encoder = pipe.text_encoder.to("cuda", dtype=torch.float16)
|
48 |
|
49 |
def can_expand(source_width, source_height, target_width, target_height, alignment):
|
50 |
"""Checks if the image can be expanded based on the alignment."""
|