Add Repository import to the FLAX example script (#12501)
This commit is contained in:
@@ -1106,6 +1106,7 @@ just three lines of code:
|
|||||||
from flax import serialization
|
from flax import serialization
|
||||||
from jax import random
|
from jax import random
|
||||||
from flax import linen as nn
|
from flax import linen as nn
|
||||||
|
from huggingface_hub import Repository
|
||||||
|
|
||||||
model = nn.Dense(features=5)
|
model = nn.Dense(features=5)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user