mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2026-02-18 12:44:29 +01:00
fix: remove "wait-for-model" header from hf request
This commit is contained in:
@@ -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"
|
||||
}
|
||||
|
||||
|
||||
@@ -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))
|
||||
|
||||
Reference in New Issue
Block a user