Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,14 +70,13 @@ pipe_controlnet = FluxControlNetPipeline(
|
|
| 70 |
tokenizer=pipe.tokenizer,
|
| 71 |
text_encoder_2=pipe.text_encoder_2,
|
| 72 |
tokenizer_2=pipe.tokenizer_2,
|
| 73 |
-
transformer=pipe.transformer, #
|
| 74 |
controlnet=controlnet,
|
| 75 |
scheduler=pipe.scheduler,
|
| 76 |
-
safety_checker=pipe.safety_checker,
|
| 77 |
-
feature_extractor=pipe.feature_extractor,
|
| 78 |
torch_dtype=dtype
|
| 79 |
).to(device)
|
| 80 |
|
|
|
|
| 81 |
MAX_SEED = 2**32 - 1
|
| 82 |
MAX_PIXEL_BUDGET = 1024 * 1024
|
| 83 |
|
|
|
|
| 70 |
tokenizer=pipe.tokenizer,
|
| 71 |
text_encoder_2=pipe.text_encoder_2,
|
| 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 |
|