add accelerate pip package and add device_map='auto'
7d126a2
Allen Park
commited on
Spaces:
Running
Running
Commit History
add transformers pip package and remove TextIteratorStreamer from imports
e303329
Allen Park
commited on
add torch
17d56be
Allen Park
commited on