aiqcamp commited on
Commit
9d60901
Β·
verified Β·
1 Parent(s): 6bda800

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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."""