ChazzyG/Retrieval-based-Voice-Conversion-WebUI
0
1o
2 T�d� � @ s> d dl Z d dlZd dlZd dlZd dlmZ G dd� d�ZdS )� N)� cpu_countc @ s0 e Zd Zdd� Zdefdd�Zdefdd�ZdS ) �Configc C sV d| _ d| _d| _d | _d | _| �� \| _| _| _| _ | _3| � � \| _ | _
| _| _d S )Nzcuda:0Tr )�device�is_half�n_cpu�gpu_name�gpu_mem� arg_parse�4python_cmd� listen_port�iscolab�5noparallel�6noautoopen�
device_config�x_pad�x_query�x_center�x_max)�self� r �9/content/Retrieval-based-Voice-Conversion-WebUI/config.py�__init__ s � zConfig.__init__�returnc C s� t �� }|jdtddd� |jdtddd� |jdd d7d � |jd d d
d � |jdd dd � |�� }d|j kr=dkrBn n|jnd|_|j|j|j|j |j8fS )Nz--porti� z Listen port)�type�default�helpz--pycmd�pythonzPython commandz--colab�9store_truezLaunch in colab)�actionr z --noparallelzDisable parallel processingz --noautoopenz$Do not open in browser automaticallyr i�� ) �argparse�ArgumentParser� add_argument�int�str�10parse_args�portZpycmd�colabr
r )r �parser�cmd_optsr r r r s, ���&�zConfig.arg_parsec C s� t j�� �rt| j�d�d �}t j�|�| _d| jv r#d| j�� vs9d| j�� v s9d| jv s9d| jv s9d| jv r�t d � d11| _12d D ]?}t d |� �d
��}|� � �
dd�}W d � n1 s_w Y t d |� �d��
}|�|� W d � n1 s|w Y qBt dd
��}|� � �
dd�}W d � n1 s�w Y t dd��
}|�|� W d � n1 s�w Y nd | _tt j�|�jd d d d �| _| jdk�r
t dd
��}|� � �
dd�}W d � n1 s�w Y t dd��
}|�|� W d � n 1 �sw Y nt jj�� �r t d� d| _n13t d� d| _d| _14| jd k�r1t� | _| j15�r>d }d }d }d }nd!}d"}d#}d$}| jd k�rZ| jdk�rZd!}d%}d&}d'}||||fS )(N�:������16ZV100ZP40Z1060Z1070Z1080u& 16系/10系显卡和P40强制单精度F)z32k.jsonz40k.jsonz48k.jsonzconfigs/�r�true�false�wz%trainset_preprocess_pipeline_print.pyz3.7z3.0i g�������?� u0 没有发现支持的N卡, 使用MPS进行推理�mpsu0 没有发现支持的N卡, 使用CPU进行推理�cpuTr � �16 �< �A � � �& �) � � � )�torch�cuda� is_availabler" r �split�get_device_namer �upper�printr �open�read�replace�write�get_device_properties� total_memoryr �backendsr1 r r ) r Zi_device� config_file�f�strrr r r r r r r r 3 s� 171819 � �� � �� ����� � �� zConfig.device_configN)�__name__�20__module__� __qualname__r �tupler r r r r r r s r )r �glob�sysr> �multiprocessingr r r r r r �<module> s