Guillermo Rodriguez
43ed6efc75
Separate cuda build in a third step
2023-01-16 00:26:04 -03:00
Guillermo Rodriguez
bc00fe35c8
Woops wrong img deseralization
2023-01-16 00:04:20 -03:00
Guillermo Rodriguez
33715ea3aa
Add img2img mode
2023-01-15 23:42:45 -03:00
Guillermo Rodriguez
f5049b91e8
Add reconnect mechanic to dgpu bus
2023-01-08 07:31:03 -03:00
Guillermo Rodriguez
509c528b45
Remove rpc heartbeat service and add it to gpu bus
2023-01-07 13:01:03 -03:00
Guillermo Rodriguez
ef71a72808
Merge pull request #3 from guilledk/heartbeat
...
Add simple heartbeat mechanic
2023-01-07 09:09:03 -03:00
Guillermo Rodriguez
c8040fe5c4
Add simple heartbeat mechanic
2023-01-07 09:06:35 -03:00
Guillermo Rodriguez
2a3778bae6
Merge pull request #2 from guilledk/fix_rectangular_imgs
...
Fix bug where we only supported square imgs on telegram frontend reply
2023-01-07 09:06:02 -03:00
Guillermo Rodriguez
1e3318b612
Fix bug where we only supported square imgs on telegram frontend reply
2023-01-07 09:05:17 -03:00
Guillermo Rodriguez
a19f75e29e
Merge pull request #1 from guilledk/protobuf
...
Switch from `msgspec` to `protobuf`
2023-01-07 06:12:07 -03:00
Guillermo Rodriguez
d0c4ca49f0
Bump version number
2023-01-06 22:36:25 -03:00
Guillermo Rodriguez
44129f5383
Forgot to add protobuf dep
2023-01-06 22:35:23 -03:00
Guillermo Rodriguez
f7aca3241e
Fix a couple bugs on telegram frontend image deseralization and sending metainfo back
2023-01-06 15:14:39 -03:00
Guillermo Rodriguez
3716b57237
Switch to protobuf
2023-01-06 14:36:50 -03:00
Guillermo Rodriguez
1d1d2fa6a1
Switch test psycopg2 dep to binary
2022-12-31 11:57:01 -03:00
Guillermo Rodriguez
bc43602af1
Fix big caption bug, and add logging
2022-12-24 15:38:40 -03:00
Guillermo Rodriguez
02c40fbb9b
Add logging
2022-12-24 14:32:17 -03:00
Guillermo Rodriguez
ee024a21b3
Disable private msging with skynet
2022-12-24 12:13:05 -03:00
Guillermo Rodriguez
f2a5c28ab1
Add say command to telegram frontend
2022-12-24 11:58:03 -03:00
Guillermo Rodriguez
d1cd7b745b
Woops forgot about redo
2022-12-24 11:54:28 -03:00
Guillermo Rodriguez
ed0e70a4ce
Fix image meta and add logging on telgram
2022-12-24 11:49:20 -03:00
Guillermo Rodriguez
050885ddbe
Fix telegram frontend bug sending upscaled images based on wrong size
2022-12-24 11:42:30 -03:00
Guillermo Rodriguez
27271f0931
Add more logging on msg passing
2022-12-24 11:34:00 -03:00
Guillermo Rodriguez
82d6512f5a
Testing unlimited msg size
2022-12-24 11:28:17 -03:00
Guillermo Rodriguez
70ca9b11cd
Need to add more cmd line params to be able to deploy
2022-12-24 11:20:25 -03:00
Guillermo Rodriguez
5652ee985a
Missing dep Pillow
2022-12-24 10:55:21 -03:00
Guillermo Rodriguez
1ec908b244
Disable torch import on cuda-less runs
2022-12-24 10:54:33 -03:00
Guillermo Rodriguez
b4289ba04f
Add upscaler
2022-12-24 10:39:40 -03:00
Guillermo Rodriguez
e5fe70d610
Drop old txt2img.py script
2022-12-21 11:59:20 -03:00
Guillermo Rodriguez
b41c2babd9
Finish telegram frontend integration
...
Correctly update user stats after a txt2img request
Swap tg_id to BIGINT on database
Refactor help topic system to be more generic
Fix skynet brain cli entry point
Fix multi request rpc session, we werent creating a context on the req side
Fix redo method
DGPU now returns image metadata
Improve error messages brain to frontend
Add version as constant
Add telegram dep to requirements
2022-12-21 11:53:50 -03:00
Guillermo Rodriguez
b255e03d65
Minor fixes to upscaler cli tool
2022-12-19 12:42:31 -03:00
Guillermo Rodriguez
6874282edf
Add authenticated messaging, also cmd line utils txt2img and upscale
2022-12-19 12:36:02 -03:00
Guillermo Rodriguez
bed5268f08
Rework dgpu client to be single task
...
Add a lot of dgpu real image gen tests
Modified docker files and environment to allow for quick test relaunch without image rebuild
Rename package from skynet_bot to skynet
Drop tractor usage cause cuda is oriented to just a single proc managing gpu resources
Add ackwnoledge phase to image request for quick gpu disconnected type scenarios
Add click entry point for dgpu
Add posibility to reuse postgres_db fixture on same session by checking if schema init has been already done
2022-12-17 11:39:42 -03:00
Guillermo Rodriguez
e7c7f97218
Add TLS for encrypted comms, next HMAC
2022-12-11 13:06:07 -03:00
Guillermo Rodriguez
4224e1645c
Remove unused import from tg module and unecesary docker command on db startup
2022-12-11 11:10:17 -03:00
Guillermo Rodriguez
f8373f681b
Started making roboust testing fixtures to init fresh db and skynet
...
Add simple dgpu worker connection test
Make db connection handler manage schema and table init logic
Keep tweaking dgpu main handler attemtping to fix subactor hangs
Change frontend open rpc logic to return a wrapped rpc_call fn referencing the new socket
Decupled user config request validation from telegram module
Fix next_worker logic, now takes in account multiple tasks per dgpu
Add dgpu_workers and dgpu_next calls
Fixed readme, moved db init code into db module
2022-12-11 11:02:55 -03:00
Guillermo Rodriguez
6247970dc8
Move gpu lifecycle tests to its own file to make torch import optional
2022-12-11 08:38:42 -03:00
Guillermo Rodriguez
20b377dd32
Add gpu worker tests to debug tractor lifecycle stuff also logging
...
Minor tweak to cuda docker image
Pin tractor branch
Add triton to cuda reqs
2022-12-11 08:32:25 -03:00
Guillermo Rodriguez
66d997c039
First decupled architecture, still working on integrating tractor gpu workers
2022-12-10 18:18:03 -03:00
Guillermo Rodriguez
f06a09b5bd
Add more ais, and start upscaler config and pipeline
2022-12-06 17:50:04 -03:00
Guillermo Rodriguez
7c1816b86e
Add 4 new ais, donation command, unknown cmd handler
2022-12-06 09:29:41 -03:00
Guillermo Rodriguez
72149ae1b4
Add roles and start working on algo selection, clean up rr code
2022-12-05 19:36:21 -03:00
Guillermo Rodriguez
31058c116a
Add round robin
2022-12-04 21:31:57 -03:00
Guillermo Rodriguez
66f21ff9de
Initial commit
2022-12-04 20:39:43 -03:00