Angelo25 commited on
Commit
7abc49e
·
verified ·
1 Parent(s): 07d7fb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ llm_pipe = pipeline(
6
  "text-generation",
7
  model="microsoft/Phi-3-mini-4k-instruct",
8
  device_map="auto", # needs `pip install accelerate`; remove to run on CPU only
9
- torch_dtype="auto",
10
- trust_remote_code=True, # Phi-3 needs this on some transformers versions
11
  )
12
  tts_pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
13
 
 
6
  "text-generation",
7
  model="microsoft/Phi-3-mini-4k-instruct",
8
  device_map="auto", # needs `pip install accelerate`; remove to run on CPU only
9
+ torch_dtype="auto"
 
10
  )
11
  tts_pipe = pipeline("text-to-speech", model="facebook/mms-tts-eng")
12