From e62baf87f8eb5b10a891816aa6e36d1ed5f6ed5b Mon Sep 17 00:00:00 2001
From: Zach Nussbaum <zanussbaum@gmail.com>
Date: Fri, 21 Apr 2023 04:19:37 +0000
Subject: [PATCH] fix: seed

---
 test_gpt_jr.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/test_gpt_jr.py b/test_gpt_jr.py
index e8c406dc..77a2c2fd 100644
--- a/test_gpt_jr.py
+++ b/test_gpt_jr.py
@@ -2,6 +2,10 @@ import torch
 from gpt4all.models import GPTJRForCausalLM, GPTJRConfig
 from transformers import AutoTokenizer, AutoModel
 
+# seed torch 
+
+torch.manual_seed(0)
+
 config = GPTJRConfig(encoder_dim=384, n_layer=4)
 print("loaded config")