theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore

Browse Source
This commit is contained in:
josc146 2023-08-16 21:11:58 +08:00
parent ce986cfc6d
commit 74f1a1c033
6 changed files with 9 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
frontend/src/components/ResetConfigsButton.tsx
View File

@ -11,7 +11,7 @@ export const ResetConfigsButton: FC<{ afterConfirm?: () => void }> = ({ afterCon
return <DialogButton icon={<ArrowReset20Regular />} tooltip={t('Reset All Configs')} title={t('Reset All Configs')} return <DialogButton icon={<ArrowReset20Regular />} tooltip={t('Reset All Configs')} title={t('Reset All Configs')}
contentText={t('Are you sure you want to reset all configs? This will obtain the latest preset configs, but will override your custom configs and cannot be undone.')} contentText={t('Are you sure you want to reset all configs? This will obtain the latest preset configs, but will override your custom configs and cannot be undone.')}
onConfirm={() => { onConfirm={() => {
commonStore.setModelConfigs(commonStore.platform != 'darwin' ? defaultModelConfigs : defaultModelConfigsMac, false); commonStore.setModelConfigs(commonStore.platform !== 'darwin' ? defaultModelConfigs : defaultModelConfigsMac, false);
commonStore.setCurrentConfigIndex(0, true); commonStore.setCurrentConfigIndex(0, true);
afterConfirm?.(); afterConfirm?.();
}} />; }} />;

10
frontend/src/pages/Configs.tsx
View File

@ -290,7 +290,7 @@ export const Configs: FC = observer(() => {
</Dropdown> </Dropdown>
} /> } />
{ {
selectedConfig.modelParameters.device != 'Custom' && <Labeled label={t('Precision')} selectedConfig.modelParameters.device !== 'Custom' && <Labeled label={t('Precision')}
desc={t('int8 uses less VRAM, but has slightly lower quality. fp16 has higher quality, and fp32 has the best quality.')} desc={t('int8 uses less VRAM, but has slightly lower quality. fp16 has higher quality, and fp32 has the best quality.')}
content={ content={
<Dropdown style={{ minWidth: 0 }} className="grow" <Dropdown style={{ minWidth: 0 }} className="grow"
@ -328,9 +328,7 @@ export const Configs: FC = observer(() => {
}} /> }} />
} /> } />
} }
{ {selectedConfig.modelParameters.device.includes('CUDA') && <div />}
selectedConfig.modelParameters.device.includes('CUDA') && <div />
}
{ {
displayStrategyImg && displayStrategyImg &&
<img style={{ width: '80vh', height: 'auto', zIndex: 100 }} <img style={{ width: '80vh', height: 'auto', zIndex: 100 }}
@ -344,7 +342,7 @@ export const Configs: FC = observer(() => {
onMouseLeave={() => setDisplayStrategyImg(false)} onMouseLeave={() => setDisplayStrategyImg(false)}
content={ content={
<Input className="grow" <Input className="grow"
placeholder={commonStore.platform != 'darwin' ? 'cuda:0 fp16 *20 -> cuda:1 fp16' : 'mps fp32'} placeholder={commonStore.platform !== 'darwin' ? 'cuda:0 fp16 *20 -> cuda:1 fp16' : 'mps fp32'}
value={selectedConfig.modelParameters.customStrategy} value={selectedConfig.modelParameters.customStrategy}
onChange={(e, data) => { onChange={(e, data) => {
setSelectedConfigModelParams({ setSelectedConfigModelParams({
@ -355,7 +353,7 @@ export const Configs: FC = observer(() => {
} }
{selectedConfig.modelParameters.device === 'Custom' && <div />} {selectedConfig.modelParameters.device === 'Custom' && <div />}
{ {
selectedConfig.modelParameters.device != 'CPU' && selectedConfig.modelParameters.device != 'MPS' && selectedConfig.modelParameters.device !== 'CPU' && selectedConfig.modelParameters.device !== 'MPS' &&
<Labeled label={t('Use Custom CUDA kernel to Accelerate')} <Labeled label={t('Use Custom CUDA kernel to Accelerate')}
desc={t('Enabling this option can greatly improve inference speed and save some VRAM, but there may be compatibility issues. If it fails to start, please turn off this option.')} desc={t('Enabling this option can greatly improve inference speed and save some VRAM, but there may be compatibility issues. If it fails to start, please turn off this option.')}
content={ content={

2
frontend/src/pages/Settings.tsx
View File

@ -126,7 +126,7 @@ export const Settings: FC = observer(() => {
} /> } />
} }
{ {
commonStore.settings.language === 'zh' && commonStore.platform != 'linux' && commonStore.settings.language === 'zh' && commonStore.platform !== 'linux' &&
<Labeled label={t('Use Tsinghua Pip Mirrors')} flex spaceBetween content={ <Labeled label={t('Use Tsinghua Pip Mirrors')} flex spaceBetween content={
<Switch checked={commonStore.settings.cnMirror} <Switch checked={commonStore.settings.cnMirror}
onChange={(e, data) => { onChange={(e, data) => {

2
frontend/src/startup.ts
View File

@ -70,7 +70,7 @@ async function initConfig() {
configData.currentModelConfigIndex >= 0 && configData.currentModelConfigIndex < configData.modelConfigs.length) configData.currentModelConfigIndex >= 0 && configData.currentModelConfigIndex < configData.modelConfigs.length)
commonStore.setCurrentConfigIndex(configData.currentModelConfigIndex, false); commonStore.setCurrentConfigIndex(configData.currentModelConfigIndex, false);
}).catch(() => { }).catch(() => {
commonStore.setModelConfigs(commonStore.platform != 'darwin' ? defaultModelConfigs : defaultModelConfigsMac, true); commonStore.setModelConfigs(commonStore.platform !== 'darwin' ? defaultModelConfigs : defaultModelConfigsMac, true);
}); });
} }

2
frontend/src/stores/commonStore.ts
View File

@ -167,7 +167,7 @@ class CommonStore {
createModelConfig = (config: ModelConfig = defaultModelConfigs[0], saveConfig: boolean = true) => { createModelConfig = (config: ModelConfig = defaultModelConfigs[0], saveConfig: boolean = true) => {
if (config.name === defaultModelConfigs[0].name) { if (config.name === defaultModelConfigs[0].name) {
// deep copy // deep copy
config = JSON.parse(JSON.stringify(commonStore.platform != 'darwin' ? defaultModelConfigs[0] : defaultModelConfigsMac[0])); config = JSON.parse(JSON.stringify(commonStore.platform !== 'darwin' ? defaultModelConfigs[0] : defaultModelConfigsMac[0]));
config.name = new Date().toLocaleString(); config.name = new Date().toLocaleString();
} }
this.modelConfigs.push(config); this.modelConfigs.push(config);

2
frontend/src/utils/index.tsx
View File

@ -373,7 +373,7 @@ export const checkDependencies = async (navigate: NavigateFunction) => {
}); });
} else { } else {
toast(depErrorMsg, { type: 'info', position: 'bottom-left' }); toast(depErrorMsg, { type: 'info', position: 'bottom-left' });
if (commonStore.platform != 'linux') if (commonStore.platform !== 'linux')
toastWithButton(t('Python dependencies are incomplete, would you like to install them?'), t('Install'), () => { toastWithButton(t('Python dependencies are incomplete, would you like to install them?'), t('Install'), () => {
InstallPyDep(commonStore.settings.customPythonPath, commonStore.settings.cnMirror).catch((e) => { InstallPyDep(commonStore.settings.customPythonPath, commonStore.settings.cnMirror).catch((e) => {
const errMsg = e.message || e; const errMsg = e.message || e;