lukecq commited on
Commit
3d2c1c1
·
verified ·
1 Parent(s): b932c5a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def load_model_processor(model_path):
12
  processor = AutoProcessor.from_pretrained(model_path)
13
  llm = LLM(
14
  model=model_path, trust_remote_code=True, gpu_memory_utilization=0.4,
15
- enforce_eager=True, device = "cpu",
16
  limit_mm_per_prompt={"audio": 5},
17
  )
18
  return llm, processor
 
12
  processor = AutoProcessor.from_pretrained(model_path)
13
  llm = LLM(
14
  model=model_path, trust_remote_code=True, gpu_memory_utilization=0.4,
15
+ enforce_eager=True, device = "cuda",
16
  limit_mm_per_prompt={"audio": 5},
17
  )
18
  return llm, processor