PrimeIntellect
/

INTELLECT-1-fp32

Text Generation

Model card Files Files and versions Community

samsja commited on Nov 29, 2024

Commit

e76fae8

•

1 Parent(s): b5c0b89

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -32,8 +32,8 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 torch.set_default_device("cuda")
-model = AutoModelForCausalLM.from_pretrained("PrimeIntellect/INTELLECT-1")
-tokenizer = AutoTokenizer.from_pretrained("PrimeIntellect/INTELLECT-1")
 input_text = "What is the Metamorphosis of Prime Intellect about?"
 input_ids = tokenizer.encode(input_text, return_tensors="pt")

 from transformers import AutoModelForCausalLM, AutoTokenizer
 torch.set_default_device("cuda")
+model = AutoModelForCausalLM.from_pretrained("PrimeIntellect/INTELLECT-1-fp32")
+tokenizer = AutoTokenizer.from_pretrained("PrimeIntellect/INTELLECT-1-fp32")
 input_text = "What is the Metamorphosis of Prime Intellect about?"
 input_ids = tokenizer.encode(input_text, return_tensors="pt")