PEEB / utils /predict.py

Commit History

cannot load to main process, load model directly at inference.
f80b905

Peijie commited on

update to support gradio 4+
92ef913

Peijie commited on

add gpu support
a410a68

Peijie commited on

dev: add special case.
66ba241

Peijie commited on

add text embeddings and change prediction logits.
395d6df

Peijie commited on

initial commit
711211a

Peijie commited on