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 |  | 

