Adding missing import
Browse files- convert.py +1 -0
convert.py
CHANGED
|
@@ -1,3 +1,4 @@
|
|
|
|
|
| 1 |
from jax import numpy as jnp
|
| 2 |
from transformers import FlaxRobertaForMaskedLM, RobertaForMaskedLM
|
| 3 |
|
| 1 |
+
import jax
|
| 2 |
from jax import numpy as jnp
|
| 3 |
from transformers import FlaxRobertaForMaskedLM, RobertaForMaskedLM
|
| 4 |
|