Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 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)
|