diff options
| author | AUTOMATIC1111 <16777216c@gmail.com> | 2023-03-11 11:45:31 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-03-11 11:45:31 +0300 |
| commit | a11ce2b96cc933ebb9e10d46603a89457ddcb9df (patch) | |
| tree | 4c9efccb08f591e3a27b7dab6e39ce4adab17bab /test | |
| parent | 0cc0ee1bcb4c24a8c9715f66cede06601bfc00c8 (diff) | |
| parent | 5fef67f6ee949a61826a3a043ea8610fd89fc371 (diff) | |
Merge pull request #7710 from space-nuko/unipc
Implement UniPC sampler
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 |