tmmdev commited on
Commit
0f530a8
·
2 Parent(s): 8d5e51b 3792cff

[update] Set writable cache directory

Browse files
Files changed (1) hide show
  1. indicator_analyzer.py +2 -2
indicator_analyzer.py CHANGED
@@ -6,8 +6,8 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
6
 
7
  class IndicatorAnalyzer:
8
  def __init__(self):
9
- self.model = AutoModelForCausalLM.from_pretrained("tmm-dev/codellama-pattern-analysis")
10
- self.tokenizer = AutoTokenizer.from_pretrained("tmm-dev/codellama-pattern-analysis")
11
 
12
  def analyze_indicators(self, ohlcv_data):
13
  indicator_prompt = f"""
 
6
 
7
  class IndicatorAnalyzer:
8
  def __init__(self):
9
+ self.model = AutoModelForCausalLM.from_pretrained("tmmdev/codellama-pattern-analysis")
10
+ self.tokenizer = AutoTokenizer.from_pretrained("tmmdev/codellama-pattern-analysis")
11
 
12
  def analyze_indicators(self, ohlcv_data):
13
  indicator_prompt = f"""