mjdousti commited on
Commit
c4b78fa
1 Parent(s): 49513bb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -0
README.md CHANGED
@@ -50,11 +50,15 @@ from transformers import LlamaTokenizer, LlamaForCausalLM
50
  import torch
51
 
52
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
 
 
53
  model = LlamaForCausalLM.from_pretrained(
54
  "universitytehran/PersianMind-v1.0",
55
  torch_dtype=torch.bfloat16,
56
  low_cpu_mem_usage=True,
57
  device_map={"": device},
 
58
  )
59
  tokenizer = LlamaTokenizer.from_pretrained(
60
  "universitytehran/PersianMind-v1.0",
 
50
  import torch
51
 
52
  device = "cuda" if torch.cuda.is_available() else "cpu"
53
+ # Generate an access token for your account as explained in
54
+ # https://huggingface.co/docs/transformers.js/guides/private
55
+ access_token = "hf_..."
56
  model = LlamaForCausalLM.from_pretrained(
57
  "universitytehran/PersianMind-v1.0",
58
  torch_dtype=torch.bfloat16,
59
  low_cpu_mem_usage=True,
60
  device_map={"": device},
61
+ token=access_token,
62
  )
63
  tokenizer = LlamaTokenizer.from_pretrained(
64
  "universitytehran/PersianMind-v1.0",