Add Repository import to the FLAX example script (#12501)

This commit is contained in:
Lysandre Debut 2021-07-05 09:51:11 +02:00 committed by GitHub
parent 2df63282e0
commit 89a8739f0c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1106,6 +1106,7 @@ just three lines of code:
from flax import serialization
from jax import random
from flax import linen as nn
from huggingface_hub import Repository
model = nn.Dense(features=5)