Upload pipeline.py
Browse files- pipeline.py +1 -0
pipeline.py
CHANGED
|
@@ -16,6 +16,7 @@
|
|
| 16 |
#
|
| 17 |
# Originally licensed under the Apache License, Version 2.0 (the "License");
|
| 18 |
# Updated to "Affero GENERAL PUBLIC LICENSE Version 3, 19 November 2007" via extensive updates to attn_mask usage.
|
|
|
|
| 19 |
|
| 20 |
from typing import Any, Dict, List, Optional, Union
|
| 21 |
|
|
|
|
| 16 |
#
|
| 17 |
# Originally licensed under the Apache License, Version 2.0 (the "License");
|
| 18 |
# Updated to "Affero GENERAL PUBLIC LICENSE Version 3, 19 November 2007" via extensive updates to attn_mask usage.
|
| 19 |
+
__all__ = ['FluxTransformer2DModelWithMasking', 'CustomPipeline']
|
| 20 |
|
| 21 |
from typing import Any, Dict, List, Optional, Union
|
| 22 |
|