highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-11-18 19:46:36 +01:00
Code Issues Packages Projects Releases Wiki Activity

unify venv calling, add basic huggingchat module

Browse Source
This commit is contained in:
Believethehype
2023-12-29 00:27:34 +01:00
parent 2c8e81d34a
commit 6de23645e4
23 changed files with 168 additions and 200 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
nostr_dvm/tasks/imageupscale.py
View File

@@ -2,7 +2,7 @@ import json
from multiprocessing.pool import ThreadPool
from nostr_dvm.backends.nova_server.utils import check_server_status, send_request_to_server
from nostr_dvm.interfaces.dvmtaskinterface import DVMTaskInterface
from nostr_dvm.interfaces.dvmtaskinterface import DVMTaskInterface, process_venv
from nostr_dvm.utils.admin_utils import AdminConfig
from nostr_dvm.utils.dvmconfig import DVMConfig, build_default_config
from nostr_dvm.utils.nip89_utils import NIP89Config, check_and_set_d_tag
@@ -129,13 +129,5 @@ def build_example(name, identifier, admin_config, server_address):
admin_config=admin_config, options=options)
def process_venv():
args = DVMTaskInterface.process_args()
dvm_config = build_default_config(args.identifier)
dvm = ImageUpscale(name="", dvm_config=dvm_config, nip89config=NIP89Config(), admin_config=None)
result = dvm.process(json.loads(args.request))
DVMTaskInterface.write_output(result, args.output)
if __name__ == '__main__':
process_venv()
process_venv(ImageUpscale)