Nebulixlabs
/

Nutral-Base

Text Generation

custom-architecture

pretrained-from-scratch

Model card Files Files and versions

Nebulixlabs commited on 29 days ago

Commit

26840d8

·

verified ·

1 Parent(s): 8fdd5b7

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -101,7 +101,7 @@ You can initialize and test this raw base model using the standard Hugging Face
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
-model_path = "YOUR_HF_USERNAME/Nutral-Base"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.float16, device_map="auto")

 from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
+model_path = "Nebullixlabs/Nutral-Base"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
 model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.float16, device_map="auto")