Fix a transformer model I missed (#47)

* Fix a transformer model I missed Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com> * Upload a new example onnx file and fix the test and tutorial (#45) * Upload a new example onnx file and fix the tutorial Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com> * Re-enable the ONNX example test Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com> --------- Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com> * Add slurm Ci test back (#46) Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com> --------- Signed-off-by: Jeremy Fowers <jeremy.fowers@amd.com> Co-authored-by: Daniel Holanda <holand.daniel@gmail.com>
onnx · Dec 1, 2023 · 3ceda8a · 3ceda8a
1 parent e77ac15
commit 3ceda8a
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/models/transformers/mpt.py b/models/transformers/mpt.py
@@ -1,9 +1,10 @@
-
 # labels: name::mpt author::transformers task::Generative_AI license::apache-2.0
 from turnkeyml.parser import parse
 from transformers import MptModel, AutoConfig
 import torch
 
+torch.manual_seed(0)
+
 # Parsing command-line arguments
 pretrained, batch_size, max_seq_length = parse(
     ["pretrained", "batch_size", "max_seq_length"]
@@ -25,4 +26,3 @@
 
 # Call model
 model(**inputs)
-