PEEB / utils

Commit History

torch version seems correct but still have gpu error.
69abaf6

Peijie commited on

directly load model weight to gpu
11e3b9b

Peijie commited on

read embeddings to gpu directly.
81867df

Peijie commited on

fix bug and tested locally.
5ac407d

Peijie commited on

Update utils/model.py
c898769
verified

chanfee commited on

Update utils/model.py
c529e7f
verified

chanfee commited on

Update utils/model.py
274d5c1
verified

chanfee commited on

Update utils/model.py
a8bc48a
verified

chanfee commited on

Update utils/model.py
a4b4452
verified

chanfee commited on

Update utils/model.py
8e66c0b
verified

chanfee commited on

Update utils/model.py
6aa6e41
verified

chanfee commited on

Update utils/model.py
1c1bbfd
verified

chanfee commited on

Update utils/model.py
bb174ad
verified

chanfee commited on

Update utils/model.py
b940e41
verified

chanfee commited on

Update utils/model.py
9705624
verified

chanfee commited on

Update utils/model.py
6b9a3f4
verified

chanfee commited on

Update utils/model.py
c806e96
verified

chanfee commited on

Update utils/model.py
0086dc0
verified

chanfee commited on

Update utils/model.py
4f0d777
verified

chanfee commited on

Update utils/model.py
48f528c
verified

chanfee commited on

Update utils/model.py
1dd67d5
verified

chanfee commited on

Update utils/model.py
8cd3456
verified

chanfee commited on

Update utils/model.py
d14f2b2
verified

chanfee commited on

Update utils/model.py
c91d22a
verified

chanfee commited on

Update utils/model.py
1eaf9da
verified

chanfee commited on

Print text_input_parts which cause cuda error.
70e6e08
verified

chanfee commited on

Print logits for debugging.
25b63fc
verified

chanfee commited on

load model in cpu first
c8636e8

Peijie commited on

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

manually install gradio
b7379dc

Peijie commited on

add deprecated gradio class
4487e5c

Peijie commited on

initial commit
711211a

Peijie commited on