Ë
    Q^(hÁ  ã                   ó€   — d dl Z d dlmZmZmZmZ d dlmZ d dlm	Z	m
Z
 d dlmZmZmZ  G d„ de«      Z G d„ d	e«      Zy)
é    N)ÚAnyÚDictÚOptionalÚUnion)ÚInferenceProviderMapping)ÚRequestParametersÚ_as_dict)ÚBaseConversationalTaskÚTaskProviderHelperÚfilter_nonec            	       óv   ‡ — e Zd Zˆ fd„Zdededefd„Zdededede	e   fd	„Z
dd
eeef   de	e   defd„Zˆ xZS )ÚHyperbolicTextToImageTaskc                 ó*   •— t         ‰| �  ddd¬«       y )NÚ
hyperbolicúhttps://api.hyperbolic.xyzztext-to-image)ÚproviderÚbase_urlÚtask)ÚsuperÚ__init__)ÚselfÚ	__class__s    €úm/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/huggingface_hub/inference/_providers/hyperbolic.pyr   z"HyperbolicTextToImageTask.__init__
   s   ø€ Ü‰Ñ ,Ð9UÐ\kÐÕló    Úmapped_modelÚapi_keyÚreturnc                  ó   — y)Nz/v1/images/generations© )r   r   r   s      r   Ú_prepare_routez(HyperbolicTextToImageTask._prepare_route   s   € Ø'r   ÚinputsÚ
parametersÚprovider_mapping_infoc                 óÂ   — |j                   }t        |«      }d|v r|j                  d«      |d<   d|v r|j                  d«      |d<   d|vrd|d<   d|vrd|d<   ||dœ|¥S )	NÚnum_inference_stepsÚstepsÚguidance_scaleÚ	cfg_scaleÚwidthi   Úheight)ÚpromptÚ
model_name)Úprovider_idr   Úpop)r   r!   r"   r#   r   s        r   Ú_prepare_payload_as_dictz2HyperbolicTextToImageTask._prepare_payload_as_dict   s…   € ð -×8Ñ8ˆÜ  Ó,ˆ
Ø  JÑ.Ø",§.¡.Ð1FÓ"GˆJ�wÑØ˜zÑ)Ø&0§n¡nÐ5EÓ&FˆJ�{Ñ#à˜*Ñ$Ø"%ˆJ�wÑØ˜:Ñ%Ø#&ˆJ�xÑ Ø °ÑKÀ
ÐKÐKr   ÚresponseÚrequest_paramsc                 óT   — t        |«      }t        j                  |d   d   d   «      S )NÚimagesr   Úimage)r	   Úbase64Ú	b64decode)r   r0   r1   Úresponse_dicts       r   Úget_responsez&HyperbolicTextToImageTask.get_response    s,   € Ü  Ó*ˆÜ×Ñ ¨hÑ 7¸Ñ :¸7Ñ CÓDÐDr   )N)Ú__name__Ú
__module__Ú__qualname__r   Ústrr    r   r   r   r   r/   r   Úbytesr   r8   Ú__classcell__©r   s   @r   r   r   	   s   ø„ ômð(¨3ð (¸ð (Àó (ðLØðLØ'+ðLØD\ðLà	�$‰óLñ E U¨5°$¨;Ñ%7ð EÈÐRcÑIdð EÐps÷ Er   r   c                   ó(   ‡ — e Zd ZdZdefˆ fd„Zˆ xZS )ÚHyperbolicTextGenerationTaskzf
    Special case for Hyperbolic, where text-generation task is handled as a conversational task.
    r   c                 ó6   •— t         ‰| �  dd¬«       || _        y )Nr   r   )r   r   )r   r   r   )r   r   r   s     €r   r   z%HyperbolicTextGenerationTask.__init__*   s$   ø€ Ü‰ÑØ!Ø1ð 	ô 	
ð ˆ�	r   )r9   r:   r;   Ú__doc__r<   r   r>   r?   s   @r   rA   rA   %   s   ø„ ñð˜S÷ ñ r   rA   )r5   Útypingr   r   r   r   Úhuggingface_hub.hf_apir   Ú!huggingface_hub.inference._commonr   r	   Ú,huggingface_hub.inference._providers._commonr
   r   r   r   rA   r   r   r   ú<module>rH      s6   ðÛ ß -Ó -å ;ß Iß pÑ pôEÐ 2ô Eô8
Ð#9õ 
r   