highperfocused/nostrdvm
1
0
Fork 0
mirror of https://github.com/believethehype/nostrdvm.git synced 2025-10-11 04:52:41 +02:00
Code Issues Packages Projects Releases Wiki Activity

cleanup

Browse Source
This commit is contained in:
Believethehype
2023-12-15 10:38:21 +01:00
parent 3dc79d7dc9
commit fed8d16258
5 changed files with 13 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
nostr_dvm/tasks/imagegeneration_sdxl.py
View File

@@ -195,9 +195,9 @@ def build_example(name, identifier, admin_config, server_address, default_model=
} }
} }
nip89config = NIP89Config() nip89config = NIP89Config()
nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
nip89info["image"])
nip89config.CONTENT = json.dumps(nip89info) nip89config.CONTENT = json.dumps(nip89info)
return ImageGenerationSDXL(name=name, dvm_config=dvm_config, nip89config=nip89config, return ImageGenerationSDXL(name=name, dvm_config=dvm_config, nip89config=nip89config,
admin_config=admin_config, options=options) admin_config=admin_config, options=options)

6
nostr_dvm/tasks/imagegeneration_sdxlimg2img.py
View File

@@ -226,11 +226,9 @@ def build_example(name, identifier, admin_config, server_address, default_lora="
options = {'default_lora': default_lora, 'strength': strength, 'server': server_address} options = {'default_lora': default_lora, 'strength': strength, 'server': server_address}
nip89config = NIP89Config() nip89config = NIP89Config()
nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY,
nip89info["image"])
nip89config.CONTENT = json.dumps(nip89info) nip89config.CONTENT = json.dumps(nip89info)
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
return ImageGenerationSDXLIMG2IMG(name=name, dvm_config=dvm_config, nip89config=nip89config, return ImageGenerationSDXLIMG2IMG(name=name, dvm_config=dvm_config, nip89config=nip89config,
admin_config=admin_config, options=options) admin_config=admin_config, options=options)

5
nostr_dvm/tasks/imageinterrogator.py
View File

@@ -137,10 +137,9 @@ def build_example(name, identifier, admin_config, server_address):
options = {'server': server_address} options = {'server': server_address}
nip89config = NIP89Config() nip89config = NIP89Config()
nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
nip89info["image"])
nip89config.CONTENT = json.dumps(nip89info) nip89config.CONTENT = json.dumps(nip89info)
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
return ImageInterrogator(name=name, dvm_config=dvm_config, nip89config=nip89config, return ImageInterrogator(name=name, dvm_config=dvm_config, nip89config=nip89config,
admin_config=admin_config, options=options) admin_config=admin_config, options=options)

12
nostr_dvm/tasks/imageupscale.py
View File

@@ -113,6 +113,9 @@ def build_example(name, identifier, admin_config, server_address):
dvm_config = build_default_config(identifier) dvm_config = build_default_config(identifier)
admin_config.LUD16 = dvm_config.LN_ADDRESS admin_config.LUD16 = dvm_config.LN_ADDRESS
# A module might have options it can be initialized with, here we set a default model, lora and the server
options = {'server': server_address}
nip89info = { nip89info = {
"name": name, "name": name,
"image": "https://image.nostr.build/229c14e440895da30de77b3ca145d66d4b04efb4027ba3c44ca147eecde891f1.jpg", "image": "https://image.nostr.build/229c14e440895da30de77b3ca145d66d4b04efb4027ba3c44ca147eecde891f1.jpg",
@@ -126,15 +129,10 @@ def build_example(name, identifier, admin_config, server_address):
} }
} }
} }
# A module might have options it can be initialized with, here we set a default model, lora and the server
options = {'server': server_address}
nip89config = NIP89Config() nip89config = NIP89Config()
nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
nip89info["image"])
nip89config.CONTENT = json.dumps(nip89info) nip89config.CONTENT = json.dumps(nip89info)
# We add an optional AdminConfig for this one, and tell the dvm to rebroadcast its NIP89
return ImageUpscale(name=name, dvm_config=dvm_config, nip89config=nip89config, return ImageUpscale(name=name, dvm_config=dvm_config, nip89config=nip89config,
admin_config=admin_config, options=options) admin_config=admin_config, options=options)

4
nostr_dvm/tasks/textextraction_whisperx.py
View File

@@ -179,9 +179,9 @@ def build_example(name, identifier, admin_config, server_address):
} }
} }
nip89config = NIP89Config() nip89config = NIP89Config()
nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89config.DTAG = check_and_set_d_tag(identifier, name, dvm_config.PRIVATE_KEY, nip89info["image"])
nip89info["image"])
nip89config.CONTENT = json.dumps(nip89info) nip89config.CONTENT = json.dumps(nip89info)
return SpeechToTextWhisperX(name=name, dvm_config=dvm_config, nip89config=nip89config, return SpeechToTextWhisperX(name=name, dvm_config=dvm_config, nip89config=nip89config,
admin_config=admin_config, options=options) admin_config=admin_config, options=options)