PEEB / utils

Commit History

  1. torch version seems correct but still have gpu error. Running

    69abaf6
  2. directly load model weight to gpu

    11e3b9b
  3. read embeddings to gpu directly.

    81867df
  4. fix bug and tested locally.

    5ac407d

  1. Update utils/model.py

    Verified signature
    c898769
  2. Update utils/model.py

    Verified signature
    c529e7f
  3. Update utils/model.py

    Verified signature
    274d5c1
  4. Update utils/model.py

    Verified signature
    a8bc48a
  5. Update utils/model.py

    Verified signature
    a4b4452
  6. Update utils/model.py

    Verified signature
    8e66c0b
  7. Update utils/model.py

    Verified signature
    6aa6e41
  8. Update utils/model.py

    Verified signature
    1c1bbfd
  9. Update utils/model.py

    Verified signature
    bb174ad
  10. Update utils/model.py

    Verified signature
    b940e41
  11. Update utils/model.py

    Verified signature
    9705624
  12. Update utils/model.py

    Verified signature
    6b9a3f4
  13. Update utils/model.py

    Verified signature
    c806e96
  14. Update utils/model.py

    Verified signature
    0086dc0
  15. Update utils/model.py

    Verified signature
    4f0d777
  16. Update utils/model.py

    Verified signature
    48f528c
  17. Update utils/model.py

    Verified signature
    1dd67d5
  18. Update utils/model.py

    Verified signature
    8cd3456
  19. Update utils/model.py

    Verified signature
    d14f2b2
  20. Update utils/model.py

    Verified signature
    c91d22a
  21. Update utils/model.py

    Verified signature
    1eaf9da
  22. Print text_input_parts which cause cuda error.

    Verified signature
    70e6e08
  23. Print logits for debugging.

    Verified signature
    25b63fc

  1. load model in cpu first

    c8636e8
  2. load the model to cpu first

    401a164
  3. Assuming GPUZero is always available.

    e1c03cb
  4. cannot load to main process, load model directly at inference.

    f80b905
  5. update to support gradio 4+

    92ef913
  6. add gpu support

    a410a68
  7. dev: add special case.

    66ba241

  1. add text embeddings and change prediction logits.

    395d6df
  2. manually install gradio

    b7379dc
  3. add deprecated gradio class

    4487e5c
  4. initial commit

    711211a