Save and loading the model
#20
by
HSohoye
- opened
Is it possible to save this model as a single .pt file. Using the model in a GCP Dataflow pipeline and calling the files from a GCS bucket, is proving difficult using the hugging face model handler. However, using the PyTorch model handler this may be feasible but we would need the model to be saved as a .pt file instead