tsukumi
|
80dd6dbc22
|
Improve: Use the device_map option in transformers to load BERT models directly to the GPU
|
2024-09-17 05:22:06 +09:00 |
|
litagin02
|
a9c5d18eec
|
Try to fix colab torch version
|
2024-09-09 16:30:44 +09:00 |
|
litagin02
|
38adf43924
|
Fix: colab torchvision bug, explicit torchaudio version
|
2024-09-09 15:31:30 +09:00 |
|
litagin02
|
7a99e3e1b7
|
Update requirements-colab.txt
|
2024-07-27 12:32:13 +09:00 |
|
litagin02
|
52bb5dc56d
|
Fix: require numpy<2 for numpy major version up
|
2024-06-17 06:54:50 +09:00 |
|
litagin02
|
5dd7db10ce
|
Require gradio>=4.32 due to render decorator
|
2024-06-16 18:45:47 +09:00 |
|
litagin02
|
287db8cf28
|
Fix: add onnxruntime to colab req
|
2024-06-01 15:18:50 +09:00 |
|
litagin02
|
e66074c243
|
Fix: add torch to req
|
2024-05-29 07:06:00 +09:00 |
|
litagin02
|
ceb431192c
|
colab
|
2024-05-28 18:37:44 +09:00 |
|