Spaces:
Sleeping
Sleeping
Vishnu-add
commited on
Commit
•
29eb8f9
1
Parent(s):
b33c6ec
Commented device in llm pipeline
Browse files
app.py
CHANGED
@@ -36,7 +36,7 @@ def llm_pipeline():
|
|
36 |
do_sample = True,
|
37 |
temperature = 0.3,
|
38 |
top_p = 0.95,
|
39 |
-
device=device
|
40 |
)
|
41 |
|
42 |
local_llm = HuggingFacePipeline(pipeline = pipe)
|
|
|
36 |
do_sample = True,
|
37 |
temperature = 0.3,
|
38 |
top_p = 0.95,
|
39 |
+
# device=device
|
40 |
)
|
41 |
|
42 |
local_llm = HuggingFacePipeline(pipeline = pipe)
|