Vishnu-add commited on
Commit
29eb8f9
1 Parent(s): b33c6ec

Commented device in llm pipeline

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)