Spaces:
Build error
Build error
Commit
•
cc887a8
1
Parent(s):
ba12792
Delete export_model.py
Browse files- export_model.py +0 -7
export_model.py
DELETED
|
@@ -1,7 +0,0 @@
|
|
| 1 |
-
import torch
|
| 2 |
-
|
| 3 |
-
# Load the model
|
| 4 |
-
model = VisionEncoderDecoderModel.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
|
| 5 |
-
|
| 6 |
-
# Save the model state dictionary
|
| 7 |
-
torch.save(model.state_dict(), "model.pth")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|