tamdiep106 commited on
Commit
41f0ab0
·
1 Parent(s): 8e133d9

update Readme 1

Browse files
Files changed (1) hide show
  1. README.md +15 -1
README.md CHANGED
@@ -10,4 +10,18 @@ datasets:
10
  language:
11
  - en
12
  - ja
13
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
10
  language:
11
  - en
12
  - ja
13
+ ---
14
+
15
+ To use in code:
16
+
17
+ ```python
18
+ from transformers import LlamaForCausalLM, LlamaTokenizer
19
+
20
+ model = LlamaForCausalLM.from_pretrained(
21
+ "tamdiep106/alpaca_lora_ja_en_emb-7b",
22
+ load_in_8bit=False,
23
+ torch_dtype=torch.float16,
24
+ device_map="auto"
25
+ )
26
+ tokenizer = LlamaTokenizer.from_pretrained("decapoda-research/llama-7b-hf")
27
+ ```