cosimoiaia commited on
Commit
5b298e9
·
1 Parent(s): 3d1e6ef

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -50,8 +50,12 @@ from transformers import LLaMATokenizer, LLaMAForCausalLM, GenerationConfig
50
  tokenizer = LLaMATokenizer.from_pretrained("cosimoiaia/Loquace-12B")
51
  model = LLaMAForCausalLM.from_pretrained(
52
  "cosimoiaia/Loquace-12B",
53
- load_in_8bit=True,
54
  device_map="auto",
 
 
 
 
55
  )
56
  ```
57
 
 
50
  tokenizer = LLaMATokenizer.from_pretrained("cosimoiaia/Loquace-12B")
51
  model = LLaMAForCausalLM.from_pretrained(
52
  "cosimoiaia/Loquace-12B",
53
+ load_in_4bit=True,
54
  device_map="auto",
55
+ quantization_config=BitsAndBytesConfig(
56
+ load_in_4bit=True,
57
+ llm_int8_has_fp16_weight=False
58
+ )
59
  )
60
  ```
61