BIBLIOKLEPT
/

Mnemosyne-64M

Text Generation

custom-architecture

hierarchical-attention

Model card Files Files and versions

BIBLIOKLEPT commited on 27 days ago

Commit

add8f69

·

verified ·

1 Parent(s): 65aa123

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -33,7 +33,7 @@ pipeline_tag: text-generation
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
-model_id = "YOUR_USERNAME/Mnemosyne-64M"
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(

 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+model_id = "BIBLIOKLEPT/Mnemosyne-64M"
 tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(