wasmdashai commited on
Commit
2175cb3
·
verified ·
1 Parent(s): 98fb8ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ def install_model(namemodel,tokenn,namemodelonxx):
9
  def convert_to_onnx(model,namemodelonxx):
10
  vocab_size = model.text_encoder.embed_tokens.weight.size(0)
11
  example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
12
- x=f"{"wasmdashai/namemodelonxx}.onnx"
13
  torch.onnx.export(
14
  model, # The model to be exported
15
  example_input, # Example input for the model
 
9
  def convert_to_onnx(model,namemodelonxx):
10
  vocab_size = model.text_encoder.embed_tokens.weight.size(0)
11
  example_input = torch.randint(0, vocab_size, (1, 100), dtype=torch.long)
12
+ x=f"{wasmdashai/namemodelonxx}.onnx"
13
  torch.onnx.export(
14
  model, # The model to be exported
15
  example_input, # Example input for the model