AyoubChLin commited on
Commit
939e397
·
verified ·
1 Parent(s): 127a0c4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
5
  # Initialize model and tokenizer
6
  model = AutoModelForCausalLM.from_pretrained(
7
  "microsoft/Phi-3.5-mini-instruct",
8
- low_cpu_mem_usage=True
9
  torch_dtype="auto",
10
  trust_remote_code=True,
11
  )
 
5
  # Initialize model and tokenizer
6
  model = AutoModelForCausalLM.from_pretrained(
7
  "microsoft/Phi-3.5-mini-instruct",
8
+ low_cpu_mem_usage=True,
9
  torch_dtype="auto",
10
  trust_remote_code=True,
11
  )