From d862954377540b5e8c3c3eaf0b3810df723bcc36 Mon Sep 17 00:00:00 2001 From: Konstantine Tsafatinos Date: Thu, 29 Jun 2023 19:51:33 -0400 Subject: [PATCH] add binary data option to enqueue --- skynet/cli.py | 26 ++++++++++++++++++-------- 1 file changed, 18 insertions(+), 8 deletions(-) diff --git a/skynet/cli.py b/skynet/cli.py index 72ab4da..70998d1 100755 --- a/skynet/cli.py +++ b/skynet/cli.py @@ -111,6 +111,7 @@ def download(): @click.option('--step', '-s', default=26) @click.option('--seed', '-S', default=None) @click.option('--upscaler', '-U', default='x4') +@click.option('--binary_data', '-b', default='') def enqueue( account: str, permission: str, @@ -127,16 +128,25 @@ def enqueue( 'dgpu', node_url, None, None) with open_cleos(node_url, key=key) as cleos: - if not kwargs['seed']: - kwargs['seed'] = random.randint(0, 10e9) - - req = json.dumps({ - 'method': 'diffuse', - 'params': kwargs - }) - binary = '' + # if not kwargs['seed']: + # kwargs['seed'] = random.randint(0, 10e9) + # + # req = json.dumps({ + # 'method': 'diffuse', + # 'params': kwargs + # }) + # binary = '' for i in range(jobs): + if not kwargs['seed']: + kwargs['seed'] = random.randint(0, 10e9) + + req = json.dumps({ + 'method': 'diffuse', + 'params': kwargs + }) + binary = kwargs['binary_data'] + res = trio.run(cleos.a_push_action, 'telos.gpu', 'enqueue',