Changing use_cuda to true

workflow/RemoveBG.json

@@ -1 +1 @@
-{"version": "0", "type": "workflow", "nodes": [{"id": "2", "position": {"x": 187.5, "y": 52.5}, "type": "step", "data": {"label": "HuggingfaceDataset", "name": "HuggingfaceDataset", "parameters": {"dataset_id": {"type": "string", "description": "The dataset ID from Huggingface", "value": "microsoft/cats_vs_dogs"}, "split": {"type": "string", "default": "train", "description": "The split of the dataset to use", "value": "train"}, "shuffle": {"type": "boolean", "default": false, "description": "Whether to shuffle the dataset", "value": true}, "log_data": {"type": "boolean", "default": true, "description": "Whether to log the outputs as JSON to the node UI", "value": false}, "image_columns": {"type": "list[string]", "description": "The columns in the dataset that contain images. This is to let Graphbook know how to display the images in the UI.", "required": false, "value": ["image"]}, "kwargs": {"type": "dict", "description": "Additional keyword arguments to pass to the dataset", "required": false, "value": []}}, "inputs": [], "outputs": ["out"], "category": "Huggingface", "isCollapsed": false}}, {"id": "3", "type": "resource", "position": {"x": 187.5, "y": 282.5}, "data": {"name": "RMBGModel", "parameters": {"model_name": {"type": "string", "default": "briaai/RMBG-1.4", "required": false, "description": "The name of the RMBG model.", "value": "briaai/RMBG-1.4"}, "use_cuda": {"type": "boolean", "default": true, "required": false, "description": "Whether to use CUDA acceleration.", "value": false}}, "category": "BackgroundRemoval", "label": "RMBGModel", "isCollapsed": false}}, {"id": "4", "type": "step", "position": {"x": 444.5, "y": 52.5}, "data": {"name": "RemoveBackground", "parameters": {"batch_size": {"type": "number", "description": "The batch size for background removal.", "default": 8, "value": 8}, "model": {"type": "resource", "description": "The model to use for background removal."}, "output_dir": {"type": "string", "description": "The directory to save the output images.", "default": "/tmp/output", "value": "/tmp/output"}}, "inputs": ["in"], "outputs": ["out"], "category": "BackgroundRemoval", "label": "RemoveBackground", "isCollapsed": false}}], "edges": [{"source": "2", "sourceHandle": "out", "target": "4", "targetHandle": "in", "data": {}, "id": "reactflow__edge-2out-4in"}, {"source": "3", "sourceHandle": "resource", "target": "4", "targetHandle": "model", "data": {}, "id": "reactflow__edge-3resource-4model"}]}
+{"version": "0", "type": "workflow", "nodes": [{"id": "2", "position": {"x": 187.5, "y": 52.5}, "type": "step", "data": {"label": "HuggingfaceDataset", "name": "HuggingfaceDataset", "parameters": {"dataset_id": {"type": "string", "description": "The dataset ID from Huggingface", "value": "microsoft/cats_vs_dogs"}, "split": {"type": "string", "default": "train", "description": "The split of the dataset to use", "value": "train"}, "shuffle": {"type": "boolean", "default": false, "description": "Whether to shuffle the dataset", "value": true}, "log_data": {"type": "boolean", "default": true, "description": "Whether to log the outputs as JSON to the node UI", "value": false}, "image_columns": {"type": "list[string]", "description": "The columns in the dataset that contain images. This is to let Graphbook know how to display the images in the UI.", "required": false, "value": ["image"]}, "kwargs": {"type": "dict", "description": "Additional keyword arguments to pass to the dataset", "required": false, "value": []}}, "inputs": [], "outputs": ["out"], "category": "Huggingface", "isCollapsed": false}}, {"id": "3", "type": "resource", "position": {"x": 187.5, "y": 282.5}, "data": {"name": "RMBGModel", "parameters": {"model_name": {"type": "string", "default": "briaai/RMBG-1.4", "required": false, "description": "The name of the RMBG model.", "value": "briaai/RMBG-1.4"}, "use_cuda": {"type": "boolean", "default": true, "required": false, "description": "Whether to use CUDA acceleration.", "value": true}}, "category": "BackgroundRemoval", "label": "RMBGModel", "isCollapsed": false}}, {"id": "4", "type": "step", "position": {"x": 444.5, "y": 52.5}, "data": {"name": "RemoveBackground", "parameters": {"batch_size": {"type": "number", "description": "The batch size for background removal.", "default": 8, "value": 8}, "model": {"type": "resource", "description": "The model to use for background removal."}, "output_dir": {"type": "string", "description": "The directory to save the output images.", "default": "/tmp/output", "value": "/tmp/output"}}, "inputs": ["in"], "outputs": ["out"], "category": "BackgroundRemoval", "label": "RemoveBackground", "isCollapsed": false}}], "edges": [{"source": "2", "sourceHandle": "out", "target": "4", "targetHandle": "in", "data": {}, "id": "reactflow__edge-2out-4in"}, {"source": "3", "sourceHandle": "resource", "target": "4", "targetHandle": "model", "data": {}, "id": "reactflow__edge-3resource-4model"}]}