Update README.md
Browse files
README.md
CHANGED
@@ -95,7 +95,7 @@ The hidden representation associated to the input batch of configurations can be
|
|
95 |
```python
|
96 |
wf = FlaxAutoModel.from_pretrained("nqs-models/j1j2_square_10x10", trust_remote_code=True, return_z=True)
|
97 |
|
98 |
-
z = wf
|
99 |
```
|
100 |
|
101 |
#### Training Hyperparameters
|
|
|
95 |
```python
|
96 |
wf = FlaxAutoModel.from_pretrained("nqs-models/j1j2_square_10x10", trust_remote_code=True, return_z=True)
|
97 |
|
98 |
+
z = wf(wf.params, samples)
|
99 |
```
|
100 |
|
101 |
#### Training Hyperparameters
|