Update README.md
Browse files
README.md
CHANGED
|
@@ -47,7 +47,6 @@ from transformers import AutoModel, AutoProcessor
|
|
| 47 |
|
| 48 |
model = AutoModel.from_pretrained("dleemiller/SwipeALot-base", trust_remote_code=True)
|
| 49 |
model.eval()
|
| 50 |
-
model.requires_grad_(False)
|
| 51 |
processor = AutoProcessor.from_pretrained("dleemiller/SwipeALot-base", trust_remote_code=True)
|
| 52 |
|
| 53 |
# Load a sample row from the dataset.
|
|
|
|
| 47 |
|
| 48 |
model = AutoModel.from_pretrained("dleemiller/SwipeALot-base", trust_remote_code=True)
|
| 49 |
model.eval()
|
|
|
|
| 50 |
processor = AutoProcessor.from_pretrained("dleemiller/SwipeALot-base", trust_remote_code=True)
|
| 51 |
|
| 52 |
# Load a sample row from the dataset.
|