Xenova HF Staff commited on
Commit
cb5341f
·
verified ·
1 Parent(s): 031541e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -78,7 +78,7 @@ torch.onnx.export(
78
  },
79
  )
80
 
81
- dummy_decoder_inputs = torch.randint(8, (4, 32, 91))
82
  torch.onnx.export(
83
  decoder,
84
  dummy_decoder_inputs,
 
78
  },
79
  )
80
 
81
+ dummy_decoder_inputs = torch.randint(8, (4, model.config.num_quantizers, 91))
82
  torch.onnx.export(
83
  decoder,
84
  dummy_decoder_inputs,