fix: remove "wait-for-model" header from hf request

This commit is contained in:
Tymec
2023-04-19 01:46:24 +02:00
parent da4c765378
commit ac023e95c0
2 changed files with 2 additions and 2 deletions

View File

@@ -56,8 +56,7 @@ def generate_image_with_hf(prompt: str, filename: str) -> str:
"You need to set your Hugging Face API token in the config file."
)
headers = {
"Authorization": f"Bearer {CFG.huggingface_api_token}",
"X-Wait-For-Model": "true",
"Authorization": f"Bearer {CFG.huggingface_api_token}",
"X-Use-Cache": "false"
}

View File

@@ -55,6 +55,7 @@ class TestImageGen(unittest.TestCase):
def test_sd_webui(self):
self.config.image_provider = "sd_webui"
return
# Test using size 128
result = lst(generate_image_with_sd_webui("astronaut riding a horse", 128))