Hi, it looks like the doc-builder only supports building Colab notebooks in PyTorch/TensorFlow at the moment. Would it be possible to extend this for Flax as well? For context, we're trying to build some notebooks for Diffusers, which uses Flax in some of the training doc examples.
cc @yiyixuxu @sgugger
|
new_lines = {"mixed": [], "pt": [], "tf": []} |
Hi, it looks like the doc-builder only supports building Colab notebooks in PyTorch/TensorFlow at the moment. Would it be possible to extend this for Flax as well? For context, we're trying to build some notebooks for Diffusers, which uses Flax in some of the training doc examples.
cc @yiyixuxu @sgugger
doc-builder/src/doc_builder/convert_to_notebook.py
Line 89 in 10fd3bb