deepcrayon
/
tinyparrot
1
0
Fork 0
Code

rm comment

main
Jeff Moe 2024-02-12 10:33:19 -07:00
parent 6ae59106a6
commit c1df967d67
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/tinyparrot/tinyparrot_llama.py
View File

@ -31,7 +31,6 @@ from tinygrad.helpers import Timing, Profiling, getenv, DEBUG, colored
from tinygrad import Device, GlobalCounters, dtypes, nn
from tinygrad.tensor import Tensor
from tinygrad.nn.state import safe_load, torch_load, load_state_dict, get_parameters
#from extra.models.llama import Transformer, convert_from_huggingface
from tinyparrot.tinyllama import Transformer, convert_from_huggingface
from sentencepiece import SentencePieceProcessor