Update hf_prefixlm_converter.py
Browse files- hf_prefixlm_converter.py +1 -0
hf_prefixlm_converter.py
CHANGED
|
@@ -6,6 +6,7 @@ Causal LM to convert it to a Prefix LM.
|
|
| 6 |
Prefix LMs accepts a `bidirectional_mask` input in `forward`
|
| 7 |
and treat the input prompt as the prefix in `generate`.
|
| 8 |
"""
|
|
|
|
| 9 |
import math
|
| 10 |
import warnings
|
| 11 |
from types import MethodType
|
|
|
|
| 6 |
Prefix LMs accepts a `bidirectional_mask` input in `forward`
|
| 7 |
and treat the input prompt as the prefix in `generate`.
|
| 8 |
"""
|
| 9 |
+
|
| 10 |
import math
|
| 11 |
import warnings
|
| 12 |
from types import MethodType
|