Hugging Face's logo

DataPilot
/
ArrowNeo-AME-4x3B-v0.1-MoE

Text Generation
Safetensors
Japanese
English
mixtral
conversational
Model card Files Community
Holy-fox commited on
Commit
fe176a9
·
verified ·
1 Parent(s): bf498bc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -10,7 +10,7 @@ datasets:
10
  - Nurture-intelligence/ins_dataset
11
  ---
12
 
13
- # DataPilot/ArrowNeo-AME-3x4B-v0.1-MoE
14
 
15
  ## overview
16
 
@@ -38,7 +38,7 @@ import torch
38
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, set_seed
39
 
40
  # モデルのロード
41
- model_name = "DataPilot/ArrowNeo-AME-3x4B-v0.1-MoE"
42
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
43
  tokenizer = AutoTokenizer.from_pretrained(model_name)
44
  chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
10
  - Nurture-intelligence/ins_dataset
11
  ---
12
 
13
+ # DataPilot/ArrowNeo-AME-4x3B-v0.1-MoE
14
 
15
  ## overview
16
 
 
38
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, set_seed
39
 
40
  # モデルのロード
41
+ model_name = "DataPilot/ArrowNeo-AME-4x3B-v0.1-MoE"
42
  model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
43
  tokenizer = AutoTokenizer.from_pretrained(model_name)
44
  chat_pipeline = pipeline("text-generation", model=model, tokenizer=tokenizer)