Dhahlan2000 commited on
Commit
96b4247
·
1 Parent(s): b701198

reduced max_new_tokens

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ model = AutoModelForCausalLM.from_pretrained(
22
  torch_dtype=torch.bfloat16,
23
  token=access_token
24
  )
25
- pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=2048)
26
 
27
  # Define functions for fetching stock data, news, and moving averages
28
  def validate_ticker(ticker):
 
22
  torch_dtype=torch.bfloat16,
23
  token=access_token
24
  )
25
+ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=512)
26
 
27
  # Define functions for fetching stock data, news, and moving averages
28
  def validate_ticker(ticker):