PEEB / utils /predict.py

Commit History

load the model to cpu first
401a164

Peijie commited on

Assuming GPUZero is always available.
e1c03cb

Peijie commited on

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