abenkbp commited on
Commit
05c511f
·
1 Parent(s): 1d0d462
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
- hf_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 = api_key
22
  )
23
 
24
  @app.route('/chat', methods=['POST'])