mirror of https://github.com/skygpu/skynet.git
22 lines
509 B
Python
22 lines
509 B
Python
import json
|
|
|
|
from skynet.constants import *
|
|
from skynet.dgpu.compute import maybe_load_model, compute_one
|
|
|
|
|
|
async def test_diffuse(inject_mockers):
|
|
model = 'skygpu/txt2img-mocker'
|
|
mode = 'diffuse'
|
|
params = {
|
|
"prompt": "Kronos God Realistic 4k",
|
|
"model": model,
|
|
"step": 21,
|
|
"width": 1024,
|
|
"height": 1024,
|
|
"seed": 168402949,
|
|
"guidance": "7.5"
|
|
}
|
|
|
|
with maybe_load_model(model, mode) as model:
|
|
compute_one(model, 0, mode, params)
|