Spaces:
Runtime error
Runtime error
Ken Sang Tang commited on
Update app.py
Browse files
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)
|