Ken Sang Tang commited on
Commit
296ceae
·
verified ·
1 Parent(s): 47d5964

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -22,7 +22,7 @@ api = REST(ALPACA_API_KEY, ALPACA_SECRET_KEY, ALPACA_BASE_URL)
22
 
23
  # Load Models
24
  dolly_tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
25
- dolly_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype.float16) # half precision
26
  dolly_model.gradient_checkpointing_enable() # memory-efficient loading
27
  print("Loading FinBERT model...")
28
  finbert_tokenizer = AutoTokenizer.from_pretrained(FINBERT_MODEL_NAME)
 
22
 
23
  # Load Models
24
  dolly_tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
25
+ dolly_model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, torch_dtype=torch.float16) # half precision
26
  dolly_model.gradient_checkpointing_enable() # memory-efficient loading
27
  print("Loading FinBERT model...")
28
  finbert_tokenizer = AutoTokenizer.from_pretrained(FINBERT_MODEL_NAME)