Fix lint error due to rebase.

xorbitsai · Feb 12, 2024 · 7dafcf1 · 7dafcf1
1 parent 4f8c2cb
commit 7dafcf1
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git a/xinference/device_utils.py b/xinference/device_utils.py
@@ -17,7 +17,6 @@
 import torch
 from typing_extensions import Literal, Union
 
-
 DeviceType = Literal["cuda", "mps", "xpu", "cpu"]
 
 

diff --git a/xinference/model/llm/pytorch/yi_vl.py b/xinference/model/llm/pytorch/yi_vl.py
@@ -24,6 +24,7 @@
 import torch
 from PIL import Image
 
+from ....model.utils import select_device
 from ....types import (
     ChatCompletion,
     ChatCompletionChoice,