Arsh014 commited on
Commit
1cfea7f
·
verified ·
1 Parent(s): b0c41b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ tokenizer = AutoTokenizer.from_pretrained(base_model_id)
13
 
14
  model = AutoModelForCausalLM.from_pretrained(
15
  base_model_id,
16
- load_in_4bit=True,
17
  torch_dtype=torch.float16,
18
  device_map="auto"
19
  )
 
13
 
14
  model = AutoModelForCausalLM.from_pretrained(
15
  base_model_id,
16
+ load_in_8bit=True,
17
  torch_dtype=torch.float16,
18
  device_map="auto"
19
  )