fantaxy commited on
Commit
406065d
·
verified ·
1 Parent(s): cf21a37

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,11 +25,11 @@ def split_image(input_image, num_splits=8):
25
  output_images.append(input_image.crop(box))
26
  return output_images
27
 
 
28
  pipe = FluxPipeline.from_pretrained(
29
  "black-forest-labs/FLUX.1-dev",
30
  torch_dtype=torch_dtype,
31
- use_safetensors=True,
32
- variant="fp16"
33
  ).to(device)
34
 
35
  MAX_SEED = np.iinfo(np.int32).max
 
25
  output_images.append(input_image.crop(box))
26
  return output_images
27
 
28
+ # 파이프라인 초기화 수정
29
  pipe = FluxPipeline.from_pretrained(
30
  "black-forest-labs/FLUX.1-dev",
31
  torch_dtype=torch_dtype,
32
+ use_safetensors=True
 
33
  ).to(device)
34
 
35
  MAX_SEED = np.iinfo(np.int32).max