Spaces:
Runtime error
Runtime error
File size: 1,087 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 |
import requests
from .template import (
APITestTemplate,
realistic_girl_face_img,
disable_in_cq,
get_model,
)
def detect_template(payload, status: int = 200):
url = APITestTemplate.BASE_URL + "controlnet/detect"
resp = requests.post(url, json=payload)
assert resp.status_code == status
if status != 200:
return
resp_json = resp.json()
assert "tensor" in resp_json
assert len(resp_json["tensor"]) == len(payload["controlnet_input_images"])
return resp_json
@disable_in_cq
def test_ipadapter_clip_api():
"""Use previously saved CLIP output in ipadapter run."""
resp = detect_template(
dict(
controlnet_input_images=[realistic_girl_face_img],
controlnet_module="ip-adapter_clip_h",
)
)
ipadapter_input = resp["tensor"]
APITestTemplate(
"test_ipadapter_clip_api",
"txt2img",
payload_overrides={},
unit_overrides={
"ipadapter_input": ipadapter_input,
"model": get_model("ip-adapter_sd15"),
},
).exec()
|