Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
add "import torch"
  • Loading branch information
boyuZh authored Dec 26, 2023
1 parent 471da2e commit d11e5ca
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions fingpt/FinGPT_Forecaster/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,7 @@ First let's load the model:
from datasets import load_dataset
from transformers import AutoTokenizer, AutoModelForCausalLM
from peft import PeftModel
import torch
base_model = AutoModelForCausalLM.from_pretrained(
Expand Down

0 comments on commit d11e5ca

Please sign in to comment.