 asagi4
|
ca6864c91d
|
Add safetensors support and fix model switching
|
2024-12-08 15:16:49 +02:00 |
|
|
asagi4
|
55d26c10c0
|
Actually fix moving the model between devices.
The model produces different output on different devices for some reason, but
this matches the behaviour of the original code.
|
2024-12-08 14:48:12 +02:00 |
|
|
asagi4
|
747ad6e21b
|
Make device configurable
|
2024-12-08 01:15:06 +02:00 |
|
|
asagi4
|
67ad24bf4b
|
Cleanup
|
2024-12-08 01:00:36 +02:00 |
|
|
asagi4
|
dfe7eb8aa4
|
Don't error out on long prompts
|
2024-12-07 18:59:48 +02:00 |
|
|
asagi4
|
45567e2ddb
|
Reformat code and reshape non-square noises so that the model can run
|
2024-12-07 15:21:45 +02:00 |
|
|
asagi4
|
a0feaaa746
|
Properly pass through seed
|
2024-12-07 14:55:37 +02:00 |
|
|
asagi4
|
08e3facde1
|
It works, ship it
|
2024-12-07 14:40:51 +02:00 |
|