riaz.somc/transformers
1
0
Fork 0
mirror of https://github.com/huggingface/transformers.git synced 2025-07-24 23:08:57 +06:00
Code Issues Actions 32 Packages Projects Releases Wiki Activity

flax.linen.apply takes state as the first param, followed by the input (#12510)

Browse Source
This commit is contained in:
Navjot 2021-07-05 07:03:14 -07:00 committed by GitHub
parent f1c81d6b92
commit eceb1042c1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/research_projects/jax-projects/README.md
View File

@ -621,7 +621,7 @@ state = model_flax.init(rng, dummy_input_ids)
and then we can do the forward pass.
```python
sequences = model_flax.apply(input_ids, state)
sequences = model_flax.apply(state, input_ids)
```
Visually, the forward pass would now be represented as passing all tensors required for the computation to the model's object: