1
0
Fork 0

memoize unchanged models, add some missing formats

This commit is contained in:
Sean Sube 2023-05-07 23:35:15 -05:00
parent 1e66ffd252
commit b2ebc95317
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 12 additions and 10 deletions

View File

@ -5,7 +5,7 @@ import { useStore } from 'zustand';
import { OnnxState, StateContext } from '../../state';
const { useContext, useState } = React;
const { useContext, useState, memo, useMemo } = React;
export interface EditableListProps<T> {
// items: Array<T>;
@ -21,21 +21,20 @@ export interface EditableListProps<T> {
}
export function EditableList<T>(props: EditableListProps<T>) {
const state = mustExist(useContext(StateContext));
const items = useStore(state, props.selector);
const { newItem, renderItem, setItem, selector } = props;
const { newItem, renderItem, setItem } = props;
const state = mustExist(useContext(StateContext));
const items = useStore(state, selector);
const [nextLabel, setNextLabel] = useState('');
const [nextSource, setNextSource] = useState('');
const RenderMemo = useMemo(() => memo(renderItem), [renderItem]);
return <Stack spacing={2}>
{items.map((model, idx) => <Stack direction='row' key={idx} spacing={2}>
{renderItem({
model,
onChange(t) {
setItem(t);
},
})}
<RenderMemo
model={model}
onChange={setItem}
/>
<Button onClick={() => {
// removeItem(model);
}}>Remove</Button>

View File

@ -45,6 +45,7 @@ export function ExtraNetworkInput(props: ExtraNetworkInputProps) {
>
<MenuItem value='ckpt'>ckpt</MenuItem>
<MenuItem value='safetensors'>safetensors</MenuItem>
<MenuItem value='bin'>bin</MenuItem>
</Select>
<Select value={model.type} label='Type' onChange={(selection) => {
onChange({

View File

@ -37,6 +37,8 @@ export function ExtraSourceInput(props: ExtraSourceInputProps) {
>
<MenuItem value='ckpt'>ckpt</MenuItem>
<MenuItem value='safetensors'>safetensors</MenuItem>
<MenuItem value='json'>json</MenuItem>
<MenuItem value='yaml'>yaml</MenuItem>
</Select>
<TextField label='Folder' value={model.dest} onChange={(event) => {
onChange({