Neural nodes progress

Author: luboslenco

paint/sources/neural_nodes/edit_image_node.ts

@@ -58,7 +58,7 @@ function edit_image_node_button(node_id: i32) {
 				dir + "/Qwen2.5-VL-7B-Instruct-Q8_0.gguf", "--qwen2vl_vision", dir + "/Qwen2.5-VL-7B-Instruct.mmproj-Q8_0.gguf", "--sampling-method", "euler",
 				"--offload-to-cpu",
 				// "--diffusion-fa",
-				"--cfg-scale", "2.5", "--flow-shift", "3", "--steps", "40", "-s", "-1", "-W", "512", "-H", "512", "-p", "'" + prompt + "'", "-r",
+				"--cfg-scale", "2.5", "--flow-shift", "3", "--steps", "20", "-s", "-1", "-W", "512", "-H", "512", "-p", "'" + prompt + "'", "-r",
 				dir + "/input.png", "-o", dir + "/output.png", null
 			];
 

paint/sources/neural_nodes/text_to_image_node.ts

@@ -36,7 +36,7 @@ function text_to_image_node_run_qwen(dir: string, prompt: string): string[] {
 		"-H",
 		"512",
 		"--steps",
-		"40",
+		"20",
 		"-s",
 		"-1",
 		"-o",
@@ -52,13 +52,13 @@ function text_to_image_node_run_wan(dir: string, prompt: string): string[] {
 	let argv: string[] = [
 		dir + "/sd", "-M", "vid_gen", "--diffusion-model", dir + "/Wan2.2-T2V-A14B-LowNoise-Q8_0.gguf", "--high-noise-diffusion-model",
 		dir + "/Wan2.2-T2V-A14B-HighNoise-Q8_0.gguf", "--vae", dir + "/wan_2.1_vae.safetensors", "--t5xxl", dir + "/umt5-xxl-encoder-Q8_0.gguf",
+		"--sampling-method", "euler", "--steps", "20",
+		"--high-noise-sampling-method", "euler", "--high-noise-steps", "10", "-W", "512", "-H", "512",
+		"--offload-to-cpu",
+		// "--flow-shift", "3.0",
 		// "--cfg-scale", "3.5",
-		"--sampling-method", "euler", "--steps", "40",
 		// "--high-noise-cfg-scale", "3.5",
-		"--high-noise-sampling-method", "euler", "--high-noise-steps", "20", "-W", "512", "-H", "512",
 		// "--diffusion-fa",
-		"--offload-to-cpu",
-		// "--flow-shift", "3.0",
 		"-s", "-1", "-o", dir + "/output.png", "-p", "'" + prompt + "'", null
 	];
 	return argv;

paint/sources/nodes_material.ts

@@ -1,7 +1,7 @@
 
 let nodes_material_categories: string[] = [
 	_tr("Input"), _tr("Texture"), _tr("Color"), _tr("Vector"), _tr("Converter"),
-	// _tr("Neural"),
+	_tr("Neural"),
 	_tr("Group")
 ];
 
@@ -91,17 +91,19 @@ function nodes_material_init() {
 	vector_math2_node_init();
 
 	nodes_material_neural = [];
-	text_to_image_node_init();
-	upscale_image_node_init();
-	edit_image_node_init();
-	// photo_to_pbr_node_init();
+	if (config_raw.experimental) {
+		text_to_image_node_init();
+		upscale_image_node_init();
+		edit_image_node_init();
+		// photo_to_pbr_node_init();
+	}
 
 	nodes_material_group = [];
 	group_node_init();
 
 	nodes_material_list = [
 		nodes_material_input, nodes_material_texture, nodes_material_color, nodes_material_vector, nodes_material_converter,
-		// nodes_material_neural,
+		nodes_material_neural,
 		nodes_material_group
 	];
 }