mirror of https://github.com/skygpu/skynet.git
				
				
				
			change push_action to a_push_action in enqueue cli
							parent
							
								
									c8471ff85b
								
							
						
					
					
						commit
						1b80de6228
					
				| 
						 | 
				
			
			@ -13,7 +13,7 @@ import asyncio
 | 
			
		|||
import requests
 | 
			
		||||
 | 
			
		||||
from leap.cleos import CLEOS
 | 
			
		||||
from leap.sugar import collect_stdout
 | 
			
		||||
from leap.sugar import collect_stdout, Name
 | 
			
		||||
from leap.hyperion import HyperionAPI
 | 
			
		||||
 | 
			
		||||
from skynet.ipfs import IPFSHTTP
 | 
			
		||||
| 
						 | 
				
			
			@ -100,6 +100,7 @@ def download():
 | 
			
		|||
    '--node-url', '-n', default='https://skynet.ancap.tech')
 | 
			
		||||
@click.option(
 | 
			
		||||
    '--reward', '-r', default='20.0000 GPU')
 | 
			
		||||
@click.option('--jobs', '-j', default=1)
 | 
			
		||||
@click.option('--algo', '-a', default='midj')
 | 
			
		||||
@click.option(
 | 
			
		||||
    '--prompt', '-p', default='a red old tractor in a sunny wheat field')
 | 
			
		||||
| 
						 | 
				
			
			@ -110,14 +111,13 @@ def download():
 | 
			
		|||
@click.option('--step', '-s', default=26)
 | 
			
		||||
@click.option('--seed', '-S', default=None)
 | 
			
		||||
@click.option('--upscaler', '-U', default='x4')
 | 
			
		||||
@click.option('--jobs', '-j', default=1)
 | 
			
		||||
@click.option('--min-verification', '-mv', default=1)
 | 
			
		||||
def enqueue(
 | 
			
		||||
    account: str,
 | 
			
		||||
    permission: str,
 | 
			
		||||
    key: str | None,
 | 
			
		||||
    node_url: str,
 | 
			
		||||
    reward: str,
 | 
			
		||||
    jobs: int,
 | 
			
		||||
    **kwargs
 | 
			
		||||
):
 | 
			
		||||
    key, account, permission = load_account_info(
 | 
			
		||||
| 
						 | 
				
			
			@ -136,12 +136,25 @@ def enqueue(
 | 
			
		|||
        })
 | 
			
		||||
        binary = ''
 | 
			
		||||
 | 
			
		||||
        for i in range(kwargs['jobs']):
 | 
			
		||||
            ec, out = cleos.push_action(
 | 
			
		||||
                'telos.gpu', 'enqueue', [account, req, binary, reward, kwargs['min-verification']], f'{account}@{permission}'
 | 
			
		||||
        for i in range(jobs):
 | 
			
		||||
            res = trio.run(cleos.a_push_action,
 | 
			
		||||
                'telos.gpu',
 | 
			
		||||
                'enqueue',
 | 
			
		||||
                {
 | 
			
		||||
                    'user': Name(account),
 | 
			
		||||
                    'request_body': req,
 | 
			
		||||
                    'binary_data': binary,
 | 
			
		||||
                    'reward': reward,
 | 
			
		||||
                    'min_verification': 1
 | 
			
		||||
                },
 | 
			
		||||
                # [account, req, binary, reward],
 | 
			
		||||
                # [account, req, binary, reward, kwargs['min_verification']],
 | 
			
		||||
                account, key, permission,
 | 
			
		||||
                #f'{account}@{permission}'
 | 
			
		||||
            )
 | 
			
		||||
            print(collect_stdout(out))
 | 
			
		||||
            assert ec == 0
 | 
			
		||||
            print(res)
 | 
			
		||||
            # print(collect_stdout(out))
 | 
			
		||||
            # assert ec == 0
 | 
			
		||||
 | 
			
		||||
@skynet.command()
 | 
			
		||||
@click.option('--loglevel', '-l', default='INFO', help='Logging level')
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue