Nicola Dall'Asen
|
4eac5b5998
|
modify example in README to use the new device and dtype function
|
2024-03-13 12:33:52 +01:00 |
|
Nicola Dall'Asen
|
40ec6491d4
|
load model on detected device and use correct dtype
|
2024-03-13 12:33:03 +01:00 |
|
Nicola Dall'Asen
|
494e622544
|
add an util function to detect platflorm and suitable dtype
|
2024-03-13 12:31:28 +01:00 |
|
Nicola Dall'Asen
|
5db8156747
|
use transformers 4.38.1 instead of 4.38.2 as autocast fails on mps on latest version
|
2024-03-13 12:30:50 +01:00 |
|
Nicola Dall'Asen
|
c99e9f1184
|
automatically detect the Python version, if above 3.10 patch collections to use collections.abc
|
2024-03-13 12:26:26 +01:00 |
|
Sinan
|
86a309683b
|
Enables image drop-in (#12)
|
2024-03-13 16:16:32 +08:00 |
|
Benjamin-eecs
|
8d4d9a6ccf
|
chore: rebase commits
|
2024-03-12 13:53:41 +08:00 |
|