Update src/pipeline.py
Browse files- src/pipeline.py +0 -2
src/pipeline.py
CHANGED
|
@@ -52,8 +52,6 @@ def load_pipeline() -> Pipeline:
|
|
| 52 |
torch_dtype=torch.bfloat16,
|
| 53 |
use_safetensors=False,
|
| 54 |
)
|
| 55 |
-
import pdb
|
| 56 |
-
pdb.set_trace()
|
| 57 |
pipeline = FluxPipeline.from_pretrained(
|
| 58 |
CHECKPOINT,
|
| 59 |
revision=REVISION,
|
|
|
|
| 52 |
torch_dtype=torch.bfloat16,
|
| 53 |
use_safetensors=False,
|
| 54 |
)
|
|
|
|
|
|
|
| 55 |
pipeline = FluxPipeline.from_pretrained(
|
| 56 |
CHECKPOINT,
|
| 57 |
revision=REVISION,
|