github/gpt4all
1
0
Fork 0
mirror of https://github.com/nomic-ai/gpt4all.git synced 2025-06-23 22:18:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: seed

Browse Source
This commit is contained in:
Zach Nussbaum 2023-04-21 04:19:37 +00:00
parent aa814757fc
commit e62baf87f8
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
test_gpt_jr.py
View File

@ -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")