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."""