Update modeling_text_encoder.py
Browse files- modeling_text_encoder.py +1 -0
modeling_text_encoder.py
CHANGED
|
@@ -7,6 +7,7 @@ from transformers import (
|
|
| 7 |
T5EncoderModel,
|
| 8 |
T5TokenizerFast,
|
| 9 |
)
|
|
|
|
| 10 |
|
| 11 |
class SD3TextEncoderWithMask(nn.Module):
|
| 12 |
def __init__(self, model_path, torch_dtype):
|
|
|
|
| 7 |
T5EncoderModel,
|
| 8 |
T5TokenizerFast,
|
| 9 |
)
|
| 10 |
+
from typing import Union, List, Optional
|
| 11 |
|
| 12 |
class SD3TextEncoderWithMask(nn.Module):
|
| 13 |
def __init__(self, model_path, torch_dtype):
|