From 9994bb67a1cc1a09f4c5eba70dfbd77c96d0ed76 Mon Sep 17 00:00:00 2001 From: hlohaus <983577+hlohaus@users.noreply.github.com> Date: Wed, 5 Feb 2025 17:07:20 +0100 Subject: Add zerogpu_uuid to demo --- g4f/Provider/hf_space/Janus_Pro_7B.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'g4f/Provider/hf_space/Janus_Pro_7B.py') diff --git a/g4f/Provider/hf_space/Janus_Pro_7B.py b/g4f/Provider/hf_space/Janus_Pro_7B.py index 0484316b..69708f13 100644 --- a/g4f/Provider/hf_space/Janus_Pro_7B.py +++ b/g4f/Provider/hf_space/Janus_Pro_7B.py @@ -70,6 +70,8 @@ class Janus_Pro_7B(AsyncGeneratorProvider, ProviderModelMixin): prompt: str = None, proxy: str = None, cookies: Cookies = None, + zerogpu_token: str = None, + zerogpu_uuid: str = None, return_conversation: bool = False, conversation: JsonConversation = None, seed: int = None, @@ -90,7 +92,8 @@ class Janus_Pro_7B(AsyncGeneratorProvider, ProviderModelMixin): session_hash = generate_session_hash() if conversation is None else getattr(conversation, "session_hash") async with StreamSession(proxy=proxy, impersonate="chrome") as session: session_hash = generate_session_hash() if conversation is None else getattr(conversation, "session_hash") - zerogpu_uuid, zerogpu_token = await get_zerogpu_token(cls.space, session, conversation, cookies) + if zerogpu_token is None: + zerogpu_uuid, zerogpu_token = await get_zerogpu_token(cls.space, session, conversation, cookies) if conversation is None or not hasattr(conversation, "session_hash"): conversation = JsonConversation(session_hash=session_hash, zerogpu_token=zerogpu_token, zerogpu_uuid=zerogpu_uuid) conversation.zerogpu_token = zerogpu_token -- cgit v1.2.3