abenkbp commited on
Commit
9ec8b78
·
1 Parent(s): 5790a1f
Files changed (1) hide show
  1. data/models/llama3-1-8b.py +1 -1
data/models/llama3-1-8b.py CHANGED
@@ -18,7 +18,7 @@ pipeline = transformers.pipeline(
18
  model=model_id,
19
  model_kwargs={"torch_dtype": torch.bfloat16},
20
  device=0,
21
- token = api_key
22
  )
23
 
24
  @app.route('/chat', methods=['POST'])
 
18
  model=model_id,
19
  model_kwargs={"torch_dtype": torch.bfloat16},
20
  device=0,
21
+ token=True
22
  )
23
 
24
  @app.route('/chat', methods=['POST'])