From ead25d33e9450013e537c232ce015b7573abde05 Mon Sep 17 00:00:00 2001
From: hlohaus <983577+hlohaus@users.noreply.github.com>
Date: Wed, 5 Feb 2025 19:37:11 +0100
Subject: Check for exception in generate of G4F

---
 g4f/Provider/hf_space/G4F.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'g4f/Provider')

diff --git a/g4f/Provider/hf_space/G4F.py b/g4f/Provider/hf_space/G4F.py
index 111d0f56..53bec3f4 100644
--- a/g4f/Provider/hf_space/G4F.py
+++ b/g4f/Provider/hf_space/G4F.py
@@ -82,6 +82,7 @@ class G4F(Janus_Pro_7B):
                 "x-zerogpu-token": zerogpu_token,
                 "x-zerogpu-uuid": zerogpu_uuid,
             }
+            headers = {k: v for k, v in headers.items() if v is not None}
             async def generate():
                 async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
                     await raise_for_status(response)
-- 
cgit v1.2.3