AdnanRiaz107 commited on
Commit
1dd1ef2
·
verified ·
1 Parent(s): 49383f2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -11,7 +11,8 @@ model = AutoModelForCausalLM.from_pretrained(
11
  "AdnanRiaz107/CodePhi-3-Mini-1K4bit",
12
  torch_dtype="auto", # Use auto for dtype selection
13
  trust_remote_code=True,
14
- attn_implementation='eager', # Keep this if you want to use 'eager'
 
15
  )
16
 
17
  # Load the tokenizer
 
11
  "AdnanRiaz107/CodePhi-3-Mini-1K4bit",
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