GenerTeam commited on
Commit
7515b17
·
verified ·
1 Parent(s): 14a1105

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -35,14 +35,14 @@ import torch
35
  from transformers import AutoTokenizer, AutoModelForCausalLM
36
 
37
  model = AutoModelForCausalLM.from_pretrained(
38
- "GenerTeam/GENERator-eukaryote-1.2b-base",
39
  attn_implementation="flash_attention_2",
40
  trust_remote_code=True,
41
  dtype=torch.bfloat16,
42
  ).cuda().eval()
43
 
44
  tokenizer = AutoTokenizer.from_pretrained(
45
- "GenerTeam/GENERator-eukaryote-1.2b-base",
46
  trust_remote_code=True,
47
  )
48
 
@@ -83,14 +83,14 @@ import torch
83
  from transformers import AutoTokenizer, AutoModelForCausalLM
84
 
85
  model = AutoModelForCausalLM.from_pretrained(
86
- "GenerTeam/GENERator-eukaryote-1.2b-base",
87
  attn_implementation="flash_attention_2",
88
  trust_remote_code=True,
89
  dtype=torch.bfloat16,
90
  ).cuda().eval()
91
 
92
  tokenizer = AutoTokenizer.from_pretrained(
93
- "GenerTeam/GENERator-eukaryote-1.2b-base",
94
  trust_remote_code=True,
95
  )
96
 
 
35
  from transformers import AutoTokenizer, AutoModelForCausalLM
36
 
37
  model = AutoModelForCausalLM.from_pretrained(
38
+ "GenerTeam/GENERator-eukaryote-3b-base",
39
  attn_implementation="flash_attention_2",
40
  trust_remote_code=True,
41
  dtype=torch.bfloat16,
42
  ).cuda().eval()
43
 
44
  tokenizer = AutoTokenizer.from_pretrained(
45
+ "GenerTeam/GENERator-eukaryote-3b-base",
46
  trust_remote_code=True,
47
  )
48
 
 
83
  from transformers import AutoTokenizer, AutoModelForCausalLM
84
 
85
  model = AutoModelForCausalLM.from_pretrained(
86
+ "GenerTeam/GENERator-eukaryote-3b-base",
87
  attn_implementation="flash_attention_2",
88
  trust_remote_code=True,
89
  dtype=torch.bfloat16,
90
  ).cuda().eval()
91
 
92
  tokenizer = AutoTokenizer.from_pretrained(
93
+ "GenerTeam/GENERator-eukaryote-3b-base",
94
  trust_remote_code=True,
95
  )
96