Chris4K commited on
Commit
051b302
1 Parent(s): 541e017

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -47,8 +47,8 @@ login(token=hf_token)
47
  # Define the model pipeline with additional generation parameters
48
  model_pipeline = pipeline(
49
  # model="meta-llama/Llama-3.2-1B",
50
- model="meta-llama/Llama-3.2-1B",
51
- pad_token_id=50256,
52
  #use_auth_token=hf_token,
53
  #max_length=1000, # You can increase this if needed
54
  max_new_tokens=900 # Limit how many tokens are generated
 
47
  # Define the model pipeline with additional generation parameters
48
  model_pipeline = pipeline(
49
  # model="meta-llama/Llama-3.2-1B",
50
+ model="dunzhang/stella_en_1.5B_v5",
51
+ #pad_token_id=50256,
52
  #use_auth_token=hf_token,
53
  #max_length=1000, # You can increase this if needed
54
  max_new_tokens=900 # Limit how many tokens are generated