diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
index fb5d256f..e0e800b9 100644
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@@ -444,13 +444,13 @@ def load_pipeline(
         if "vae_decoder_session" in components:
             pipe.vae_decoder = ORTModelVaeDecoder(
                 components["vae_decoder_session"],
-                pipe,  # TODO: find the right class to provide here. ORTModel is missing the dict json method
+                pipe,
             )
 
         if "vae_encoder_session" in components:
             pipe.vae_encoder = ORTModelVaeEncoder(
                 components["vae_encoder_session"],
-                pipe,  # TODO: find the right class to provide here. ORTModel is missing the dict json method
+                pipe,
             )
 
         if not server.show_progress:
diff --git a/gui/src/components/control/VariableControl.tsx b/gui/src/components/control/VariableControl.tsx
index a4752691..09340b08 100644
--- a/gui/src/components/control/VariableControl.tsx
+++ b/gui/src/components/control/VariableControl.tsx
@@ -19,7 +19,7 @@ export function VariableControl(props: VariableControlProps) {
   const grid = useStore(store, props.selectGrid);
 
   const stack = [
-    <Stack direction='row' spacing={2}>
+    <Stack direction='row' spacing={2} key='variable-enable'>
       <FormControl>
         <FormControlLabel
           label='Grid Mode'
@@ -36,7 +36,7 @@ export function VariableControl(props: VariableControlProps) {
 
   if (grid.enabled) {
     stack.push(
-      <Stack direction='row' spacing={2}>
+      <Stack direction='row' spacing={2} key='variable-row'>
         <FormControl>
           <InputLabel id='TODO'>Columns</InputLabel>
           <Select onChange={(event) => props.setGrid({
@@ -57,7 +57,7 @@ export function VariableControl(props: VariableControlProps) {
           },
         })} />
       </Stack>,
-      <Stack direction='row' spacing={2}>
+      <Stack direction='row' spacing={2} key='variable-column'>
         <FormControl>
           <InputLabel id='TODO'>Rows</InputLabel>
           <Select onChange={(event) => props.setGrid({
@@ -94,8 +94,8 @@ export function rangeSplit(parameter: string, value: string): Array<number | str
   return csv.flatMap((it) => expandRanges(it));
 }
 
-export const EXPR_STRICT_NUMBER = /^-?[0-9]+$/;
-export const EXPR_NUMBER_RANGE = /^([0-9]+)-([0-9]+)$/;
+export const EXPR_STRICT_NUMBER = /^-?\d+$/;
+export const EXPR_NUMBER_RANGE = /^(\d+)-(\d+)$/;
 
 export function expandRanges(range: string): Array<string | number> {
   if (EXPR_STRICT_NUMBER.test(range)) {
diff --git a/onnx-web.code-workspace b/onnx-web.code-workspace
index 71b1e537..f859a141 100644
--- a/onnx-web.code-workspace
+++ b/onnx-web.code-workspace
@@ -103,6 +103,7 @@
 			"virtualenv",
 			"VRAM",
 			"webp",
+			"xformers",
 			"zustand"
 		]
 	}