Spaces:
Running
Running
Commit
·
8c59ba3
1
Parent(s):
20d39a9
added max new tokens
Browse files
app.py
CHANGED
@@ -47,7 +47,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
47 |
torch_dtype=torch.bfloat16,
|
48 |
token = access_token
|
49 |
)
|
50 |
-
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
51 |
|
52 |
def validate_ticker(ticker):
|
53 |
# Ensure ticker is uppercase and length is reasonable (1-5 characters)
|
|
|
47 |
torch_dtype=torch.bfloat16,
|
48 |
token = access_token
|
49 |
)
|
50 |
+
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, max_new_tokens=1048)
|
51 |
|
52 |
def validate_ticker(ticker):
|
53 |
# Ensure ticker is uppercase and length is reasonable (1-5 characters)
|