omniverse1 commited on
Commit
5037537
·
verified ·
1 Parent(s): 96a2409

update app

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -3,7 +3,7 @@ import yfinance as yf
3
  import pandas as pd
4
  import numpy as np
5
  import torch
6
- from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
7
  from datetime import datetime, timedelta
8
  import plotly.graph_objects as go
9
  import plotly.express as px
@@ -30,15 +30,16 @@ from config import IDX_STOCKS, TECHNICAL_INDICATORS, PREDICTION_CONFIG
30
  @spaces.GPU(duration=120)
31
  def load_model():
32
  """Load the Amazon Chronos-Bolt model for time series forecasting"""
33
- # Using AutoModelForSeq2SeqLM for T5-based model, with trust_remote_code=True
34
  model = AutoModelForSeq2SeqLM.from_pretrained(
35
  "amazon/chronos-bolt-base",
36
  torch_dtype=torch.bfloat16,
37
  device_map="auto",
38
  trust_remote_code=True
39
  )
40
- # Adding use_fast=False to bypass the problematic tokenizer conversion
41
- tokenizer = AutoTokenizer.from_pretrained(
 
42
  "amazon/chronos-bolt-base",
43
  trust_remote_code=True,
44
  use_fast=False
 
3
  import pandas as pd
4
  import numpy as np
5
  import torch
6
+ from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, T5Tokenizer
7
  from datetime import datetime, timedelta
8
  import plotly.graph_objects as go
9
  import plotly.express as px
 
30
  @spaces.GPU(duration=120)
31
  def load_model():
32
  """Load the Amazon Chronos-Bolt model for time series forecasting"""
33
+ # FIX 1: Use AutoModelForSeq2SeqLM and trust_remote_code=True for T5-based model
34
  model = AutoModelForSeq2SeqLM.from_pretrained(
35
  "amazon/chronos-bolt-base",
36
  torch_dtype=torch.bfloat16,
37
  device_map="auto",
38
  trust_remote_code=True
39
  )
40
+ # FIX 2: Explicitly use T5Tokenizer (slow version for T5 architecture)
41
+ # and set trust_remote_code=True and use_fast=False to bypass problematic AutoTokenizer conversion
42
+ tokenizer = T5Tokenizer.from_pretrained(
43
  "amazon/chronos-bolt-base",
44
  trust_remote_code=True,
45
  use_fast=False