Update README.md
Browse files
README.md
CHANGED
|
@@ -92,8 +92,6 @@ model = AutoModel.from_pretrained(repo_id, trust_remote_code=True)
|
|
| 92 |
### 2. Bayesian inference
|
| 93 |
To obtain probabilistic outputs and uncertainty metrics, use the `mc_forward` method:
|
| 94 |
```python
|
| 95 |
-
import torch
|
| 96 |
-
|
| 97 |
# Perform 50 stochastic passes
|
| 98 |
N_SAMPLES = 50
|
| 99 |
model.eval()
|
|
|
|
| 92 |
### 2. Bayesian inference
|
| 93 |
To obtain probabilistic outputs and uncertainty metrics, use the `mc_forward` method:
|
| 94 |
```python
|
|
|
|
|
|
|
| 95 |
# Perform 50 stochastic passes
|
| 96 |
N_SAMPLES = 50
|
| 97 |
model.eval()
|