update test

This commit is contained in:
Qing 2024-02-10 12:37:10 +08:00
parent ec2db92ad9
commit 97f9e12f7e

View File

@ -25,7 +25,7 @@ def test_instruct_pix2pix(device, disable_nsfw, cpu_offload):
cfg = get_config(
strategy=HDStrategy.ORIGINAL,
prompt="What if it were snowing?",
p2p_steps=sd_steps,
sd_steps=sd_steps
)
name = f"device_{device}_disnsfw_{disable_nsfw}_cpu_offload_{cpu_offload}"