Chris4K commited on
Commit
0a9f192
1 Parent(s): 89beada

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -47,6 +47,7 @@ login(token=hf_token)
47
  model_pipeline = pipeline(
48
  # model="meta-llama/Llama-3.2-1B",
49
  model="meta-llama/Llama-3.2-1B",
 
50
  #use_auth_token=hf_token,
51
  #max_length=1000, # You can increase this if needed
52
  max_new_tokens=900 # Limit how many tokens are generated
 
47
  model_pipeline = pipeline(
48
  # model="meta-llama/Llama-3.2-1B",
49
  model="meta-llama/Llama-3.2-1B",
50
+ pad_token_id=50256,
51
  #use_auth_token=hf_token,
52
  #max_length=1000, # You can increase this if needed
53
  max_new_tokens=900 # Limit how many tokens are generated