AdnanRiaz107 commited on
Commit
0f74efe
·
verified ·
1 Parent(s): 07a30aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ model = AutoModelForCausalLM.from_pretrained(
12
  torch_dtype="auto", # Use auto for dtype selection
13
  trust_remote_code=True,
14
  attn_implementation='eager',
15
- load_in_4bit= True,# Keep this if you want to use 'eager'
16
  )
17
 
18
  # Load the tokenizer
 
12
  torch_dtype="auto", # Use auto for dtype selection
13
  trust_remote_code=True,
14
  attn_implementation='eager',
15
+ load_in_8bit= True,# Keep this if you want to use 'eager'
16
  )
17
 
18
  # Load the tokenizer