pankajmathur
/

orca_mini_3b

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Pankaj Mathur commited on Jun 24, 2023

Commit

7d050a0

•

1 Parent(s): b8b91c6

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -56,7 +56,7 @@ import torch
 from transformers import LlamaForCausalLM, LlamaTokenizer
 # Hugging Face model_path
-model_path = 'psmathur/wizardlm_alpaca_dolly_orca_open_llama_3b'
 tokenizer = LlamaTokenizer.from_pretrained(model_path)
 model = LlamaForCausalLM.from_pretrained(
  model_path, torch_dtype=torch.float16, device_map='auto',

 from transformers import LlamaForCausalLM, LlamaTokenizer
 # Hugging Face model_path
+model_path = 'psmathur/orca_mini_3b'
 tokenizer = LlamaTokenizer.from_pretrained(model_path)
 model = LlamaForCausalLM.from_pretrained(
  model_path, torch_dtype=torch.float16, device_map='auto',