Spaces:
Running
on
T4
Running
on
T4
Update langchain_transformers.py
Browse files
langchain_transformers.py
CHANGED
@@ -60,9 +60,9 @@ class ChatLLM(LLM):
|
|
60 |
def __init__(self):
|
61 |
super().__init__()
|
62 |
|
63 |
-
|
64 |
def from_model_id(
|
65 |
-
|
66 |
model_id,
|
67 |
device_map: Optional[Dict[str, int]] = None
|
68 |
):
|
|
|
60 |
def __init__(self):
|
61 |
super().__init__()
|
62 |
|
63 |
+
|
64 |
def from_model_id(
|
65 |
+
self,
|
66 |
model_id,
|
67 |
device_map: Optional[Dict[str, int]] = None
|
68 |
):
|