Spaces:
Running
Running
Update app.py
Browse files
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 |
|