Commit History

add device=cuda; initialize tokenizer and model outside of function call; add inputs to device
60ffe71

Allen Park commited on

pass inputs.input_ids not just inputs
b5e9a85

Allen Park commited on

add accelerate pip package and add device_map='auto'
7d126a2

Allen Park commited on

add 3 inputs for function and rename from greet to model_call
9d86cbe

Allen Park commited on

add transformers pip package and remove TextIteratorStreamer from imports
e303329

Allen Park commited on

add torch
17d56be

Allen Park commited on

make PROMPT not hardcoded and take input values from gradio interface
090dd00

Allen Park commited on

add .gitignore
0e15888

Allen Park commited on

add code to call lynx model
210b40c

Allen Park commited on

add application file
d232ed1

Allen Park commited on

initial commit
248e51a
verified

allenpark commited on