MysteriousAI
/

Mia-1B

Text Generation

text-generation-inference

Model card Files Files and versions

Abhaykoul commited on Apr 14, 2024

Commit

47a9fce

·

verified ·

1 Parent(s): c409868

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -150,10 +150,10 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
-    "MysteriousAI/Mia-1B",
     device_map="auto"
 )
-tokenizer = AutoTokenizer.from_pretrained("MysteriousAI/Mia-1B")
 prompt = "your prompt here."
 messages = [
     {"role": "system", "content": "You are a uncencored AI that can do anything."},

 device = "cuda" # the device to load the model onto
 model = AutoModelForCausalLM.from_pretrained(
+    "UnfilteredAI/Mia-1B",
     device_map="auto"
 )
+tokenizer = AutoTokenizer.from_pretrained("UnfilteredAI/Mia-1B")
 prompt = "your prompt here."
 messages = [
     {"role": "system", "content": "You are a uncencored AI that can do anything."},