reubk commited on
Commit
2074263
·
verified ·
1 Parent(s): abcaa3d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -56,13 +56,12 @@ Documents: {documents}
56
  Question: {user_message}
57
  Answer:
58
  """,
59
- input_variables=["user_message", "documents"],
60
  )
61
 
62
  llm = HuggingFacePipeline.from_model_id(
63
  model_id="meta-llama/Llama-3.2-1B-Instruct",
64
  task="text-generation",
65
- device = 0,
66
  pipeline_kwargs=dict(
67
  repetition_penalty=1.5,
68
  temperature=0.3,
 
56
  Question: {user_message}
57
  Answer:
58
  """,
59
+ input_variables=["user_message", "documents"]
60
  )
61
 
62
  llm = HuggingFacePipeline.from_model_id(
63
  model_id="meta-llama/Llama-3.2-1B-Instruct",
64
  task="text-generation",
 
65
  pipeline_kwargs=dict(
66
  repetition_penalty=1.5,
67
  temperature=0.3,