brendon-ai commited on
Commit
af21bfe
·
verified ·
1 Parent(s): 2ea9adc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -45,7 +45,7 @@ async def load_model():
45
  try:
46
  generator = pipeline(
47
  'text-generation',
48
- model='Qwen/Qwen2-7B-Instruct',
49
  device=device,
50
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
51
  )
 
45
  try:
46
  generator = pipeline(
47
  'text-generation',
48
+ model='microsoft/Phi-3-mini-4k-instruct',
49
  device=device,
50
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
51
  )