CreatorJarvis commited on
Commit
9863532
·
verified ·
1 Parent(s): c17c199

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def _make_pipe(model_id: str, device_type: str):
34
  "image-text-to-text",
35
  model=model_id,
36
  device=device_arg,
37
- dtype=dtype,
38
  )
39
  model = getattr(pipe, "model", None)
40
  generation_config = getattr(model, "generation_config", None)
 
34
  "image-text-to-text",
35
  model=model_id,
36
  device=device_arg,
37
+ torch_dtype=dtype,
38
  )
39
  model = getattr(pipe, "model", None)
40
  generation_config = getattr(model, "generation_config", None)