Reorganize repo (#8580)
* Put models in subfolders * Styling * Fix imports in tests * More fixes in test imports * Sneaky hidden imports * Fix imports in doc files * More sneaky imports * Finish fixing tests * Fix examples * Fix path for copies * More fixes for examples * Fix dummy files * More fixes for example * More model import fixes * Is this why you're unhappy GitHub? * Fix imports in conver command
This commit is contained in:
@@ -6,9 +6,9 @@ from transformers.testing_utils import require_flax, slow
|
||||
|
||||
if is_flax_available():
|
||||
import jax
|
||||
from transformers.modeling_flax_auto import FlaxAutoModel
|
||||
from transformers.modeling_flax_bert import FlaxBertModel
|
||||
from transformers.modeling_flax_roberta import FlaxRobertaModel
|
||||
from transformers.models.auto.modeling_flax_auto import FlaxAutoModel
|
||||
from transformers.models.bert.modeling_flax_bert import FlaxBertModel
|
||||
from transformers.models.roberta.modeling_flax_roberta import FlaxRobertaModel
|
||||
|
||||
|
||||
@require_flax
|
||||
|
||||
Reference in New Issue
Block a user