diff options
| author | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-10 03:30:20 -0800 |
|---|---|---|
| committer | space-nuko <24979496+space-nuko@users.noreply.github.com> | 2023-02-10 03:30:20 -0800 |
| commit | 125319988984987801dc4b4ab1e5ed36e9b211c5 (patch) | |
| tree | 075923068ff40724e27b4cfd4ebd13b22f0bae84 /test | |
| parent | ea9bd9fc7409109adcd61b897abc2c8881161256 (diff) | |
Working UniPC (for batch size 1)
Diffstat (limited to 'test')
| -rw-r--r-- | test/basic_features/txt2img_test.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/test/basic_features/txt2img_test.py b/test/basic_features/txt2img_test.py index 5aa43a44..cb525fbb 100644 --- a/test/basic_features/txt2img_test.py +++ b/test/basic_features/txt2img_test.py @@ -66,6 +66,8 @@ class TestTxt2ImgWorking(unittest.TestCase): self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) self.simple_txt2img["sampler_index"] = "DDIM" self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) + self.simple_txt2img["sampler_index"] = "UniPC" + self.assertEqual(requests.post(self.url_txt2img, json=self.simple_txt2img).status_code, 200) def test_txt2img_multiple_batches_performed(self): self.simple_txt2img["n_iter"] = 2 |