Spaces:
Runtime error
Runtime error
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."""
|