Isaak-Carter commited on
Commit
6593855
1 Parent(s): 485bfef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -15,7 +15,10 @@ Messages = List[Dict[str, str]]
15
  # Load model and tokenizer
16
  model_name = 'mlx-community/J.O.S.I.E.3-Beta12-7B-slerp-8-bit'
17
  tokenizer = AutoTokenizer.from_pretrained(model_name)
18
- model = AutoModelForCausalLM.from_pretrained(model_name)
 
 
 
19
 
20
  def clear_session() -> History:
21
  return '', []
 
15
  # Load model and tokenizer
16
  model_name = 'mlx-community/J.O.S.I.E.3-Beta12-7B-slerp-8-bit'
17
  tokenizer = AutoTokenizer.from_pretrained(model_name)
18
+ model = AutoModelForCausalLM.from_pretrained(
19
+ model_name,
20
+ ignore_mismatched_sizes=True,
21
+ )
22
 
23
  def clear_session() -> History:
24
  return '', []