wasmdashai commited on
Commit
a509b65
·
verified ·
1 Parent(s): 138ab6b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -97,7 +97,7 @@ class OnnxModelConverter:
97
  onnx_file = f"/tmp/{onnx_filename}.onnx"
98
 
99
  vocab_size = model.text_encoder.embed_tokens.weight.size(0)
100
- example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.LongTensor)
101
 
102
  torch.onnx.export(
103
  model,
 
97
  onnx_file = f"/tmp/{onnx_filename}.onnx"
98
 
99
  vocab_size = model.text_encoder.embed_tokens.weight.size(0)
100
+ example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
101
 
102
  torch.onnx.export(
103
  model,