openfree commited on
Commit
8990a11
·
verified ·
1 Parent(s): e555cbd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -72,11 +72,12 @@ pipe_controlnet = FluxControlNetPipeline(
72
  tokenizer_2=pipe.tokenizer_2,
73
  transformer=pipe.transformer, # unet 대신 transformer 사용
74
  controlnet=controlnet,
75
- scheduler=pipe.scheduler,
76
- torch_dtype=dtype
77
  ).to(device)
78
 
79
 
 
 
80
  MAX_SEED = 2**32 - 1
81
  MAX_PIXEL_BUDGET = 1024 * 1024
82
 
 
72
  tokenizer_2=pipe.tokenizer_2,
73
  transformer=pipe.transformer, # unet 대신 transformer 사용
74
  controlnet=controlnet,
75
+ scheduler=pipe.scheduler
 
76
  ).to(device)
77
 
78
 
79
+
80
+
81
  MAX_SEED = 2**32 - 1
82
  MAX_PIXEL_BUDGET = 1024 * 1024
83