omniverse1 commited on
Commit
4fc6d0b
·
verified ·
1 Parent(s): 649d480

update app

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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,19 +30,19 @@ 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
- # 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: Use AutoTokenizer with a combination of flags to bypass the problematic tiktoken conversion logic
41
- tokenizer = AutoTokenizer.from_pretrained(
 
42
  "amazon/chronos-bolt-base",
43
  trust_remote_code=True,
44
- use_fast=False,
45
- force_download=True
46
  )
47
  return model, tokenizer
48
 
 
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 (correct 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) with trust_remote_code=True and use_fast=False
41
+ # This bypasses the faulty AutoTokenizer/fast conversion logic.
42
+ tokenizer = T5Tokenizer.from_pretrained(
43
  "amazon/chronos-bolt-base",
44
  trust_remote_code=True,
45
+ use_fast=False
 
46
  )
47
  return model, tokenizer
48