File tree Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Expand file tree Collapse file tree 2 files changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -31,7 +31,7 @@ def __init__(
3131 use_kv_cache : bool = False ,
3232 ):
3333 device = "cuda" if torch .cuda .is_available () else "cpu"
34- super ().__init__ (device = device )
34+ super ().__init__ (device = device , pretrained = "gpt2" )
3535 self ._model = model
3636 self ._tokenizer = tokenizer
3737 self ._device = torch .device (device )
Original file line number Diff line number Diff line change @@ -15,7 +15,7 @@ pip install --no-use-pep517 "git+https://github.com/pytorch/ao.git@${TORCHAO_VER
1515
1616# Install lm-eval for Model Evaluation with lm-evalution-harness
1717# Install tiktoken for tokenizer
18- pip install lm_eval==0.4.2
18+ pip install lm_eval==0.4.5
1919pip install tiktoken blobfile
2020
2121# Call the install helper for further setup
You can’t perform that action at this time.
0 commit comments