o
    ´©i  ã                   @   sn   d Z ddlZddlmZ e ¡ d ZdedB defdd„ZG d	d
„ d
ƒZ	e	ƒ Z
e
jr3e
jdus5J dƒ‚dS dS )Ú
é    N)ÚPathz.zerogpuÚvalueÚreturnc                 C   s   | d uo	|   ¡ dv S )N)Ú1ÚtÚtrue)Úlower)r   © r
   úA/home/ubuntu/.local/lib/python3.10/site-packages/spaces/config.pyÚboolean
   s   r   c                   @   s   e Zd Zdd„ ZdS )ÚSettingsc                 C   sÞ   t t d¡ƒ| _t d¡| _t t d¡ƒ| _t t d¡ƒ| _t t dd¡ƒ| _t dtt	d ƒ¡| _
t d	d
¡| _t dd¡| _tt dd¡ƒ| _tt dd¡ƒ| _tt dd¡ƒ| _tt dd¡ƒ| _tt dd¡ƒ| _d S )NÚSPACES_ZERO_GPUÚSPACES_ZERO_DEVICE_API_URLÚSPACES_GRADIO_AUTO_WRAPÚZERO_GPU_PATCH_TORCH_DEVICEÚ
ZEROGPU_V2r   ÚZEROGPU_OFFLOAD_DIRÚtensorsÚZEROGPU_PROC_SELF_CGROUP_PATHz/proc/self/cgroupÚZEROGPU_CUDA_DEVICE_NAMEzNVIDIA H200 MIG 3g.71gbÚZEROGPU_CUDA_TOTAL_MEMORYl      @E ÚZEROGPU_CUDA_RESERVED_MEMORYr   ÚZEROGPU_CUDA_CAPABILITY_MAJORé	   ÚZEROGPU_CUDA_CAPABILITY_MINORÚ"ZEROGPU_CUDA_MULTI_PROCESSOR_COUNTé<   )r   ÚosÚgetenvÚzero_gpuÚzero_device_api_urlÚgradio_auto_wrapÚzero_patch_torch_deviceÚzero_gpu_v2ÚstrÚZEROGPU_HOMEÚzerogpu_offload_dirÚzerogpu_proc_self_cgroup_pathÚzerogpu_cuda_device_nameÚintÚzerogpu_cuda_total_memoryÚzerogpu_cuda_reserved_memoryÚzerogpu_cuda_capability_majorÚzerogpu_cuda_capability_minorÚ"zerogpu_cuda_multi_processor_count)Úselfr
   r
   r   Ú__init__   sF   ÿÿÿÿ
ÿÿ
ÿ
ÿ
ÿ
ÿ
ÿ
ÿ

ÿzSettings.__init__N)Ú__name__Ú
__module__Ú__qualname__r1   r
   r
   r
   r   r      s    r   zaSPACES_ZERO_DEVICE_API_URL env must be set on ZeroGPU Spaces (identified by SPACES_ZERO_GPU=true))Ú__doc__r   Úpathlibr   Úhomer&   r%   Úboolr   r   ÚConfigr    r!   r
   r
   r
   r   Ú<module>   s    ÿÿ