nodes/Runware-ComfyUI-Runware
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #167 from Runware/feature-trellis-dracoCompression

Browse Source 
Added Support for Draco Compression in 3D Inference
This commit is contained in:
Sirshendu Ganguly
2026-06-09 17:56:26 +05:30
committed by GitHub
parent 0bfe164c32 a793c18978
commit 00af55528d
10 changed files with 316 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
__init__.py
+5
View File
@@ -122,12 +122,14 @@ from .modules.advancedFeatureSettings.wanAnimateAdvancedFeatureSettings import R
from .modules.saveImage import RunwareSaveImage
from .modules.videoOutputs import RunwareVideoInferenceOutputs
from .modules.threeDInference import threeDInference
from .modules.threeDModelSearch import Runware3DModelSearch
from .modules.threeDInferenceInputs import threeDInferenceInputs
from .modules.threeDInferenceSettings import Runware3DInferenceSettings
from .modules.threeDInferenceSettingsSparseStructure import Runware3DInferenceSettingsSparseStructure
from .modules.threeDInferenceSettingsShapeSlat import Runware3DInferenceSettingsShapeSlat
from .modules.threeDInferenceSettingsTexSlat import Runware3DInferenceSettingsTexSlat
from .modules.threeDInferenceSettingsMeshCluster import Runware3DInferenceSettingsMeshCluster
from .modules.threeDInferenceSettingsDracoCompression import Runware3DInferenceSettingsDracoCompression
from .modules.save3D import RunwareSave3D
from .modules.saveText import RunwareSaveText
from .modules.loadMesh import RunwareLoadMesh
@@ -267,12 +269,14 @@ NODE_CLASS_MAPPINGS = {
"Runware Save Image": RunwareSaveImage,
"Runware Video Inference Outputs": RunwareVideoInferenceOutputs,
"Runware 3D Inference": threeDInference,
"Runware 3D Model Search": Runware3DModelSearch,
"Runware 3D Inference Inputs": threeDInferenceInputs,
"Runware 3D Inference Settings": Runware3DInferenceSettings,
"Runware 3D Inference Settings Sparse Structure": Runware3DInferenceSettingsSparseStructure,
"Runware 3D Inference Settings Shape Slat": Runware3DInferenceSettingsShapeSlat,
"Runware 3D Inference Settings Tex Slat": Runware3DInferenceSettingsTexSlat,
"Runware 3D Inference Settings Mesh Cluster": Runware3DInferenceSettingsMeshCluster,
"Runware 3D Inference Settings Draco Compression": Runware3DInferenceSettingsDracoCompression,
"Runware Save 3D": RunwareSave3D,
"Runware Save Text": RunwareSaveText,
"Runware Load Mesh": RunwareLoadMesh,
@@ -286,6 +290,7 @@ NODE_DISPLAY_NAME_MAPPINGS = {
"Runware VAE Search": "Runware VAE",
"Runware Multi Inference": "Runware Multi Inference [BETA]",
"Runware Video Model Search": "Runware Video Model",
"Runware 3D Model Search": "Runware 3D Model",
"Runware Audio Model Search": "Runware Audio Model",
"Runware Text Model Search": "Runware Text Model",
"Runware Image Inference Settings Color Palette": "Runware Image Inference Color Palette",
clientlibs/main.js
+5 -1
View File
@@ -1,6 +1,6 @@
import { app } from "../../scripts/app.js";
import { api } from "../../scripts/api.js";
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, saveTextHandler, mediaUUIDHandler, save3DFilepathHandler, videoTranscriptionHandler, videoOutputsHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, audioModelSearchFilterHandler, textModelSearchFilterHandler, vectorizeModelSearchFilterHandler, vectorizeToggleHandler, useParameterToggleHandler, imageInferenceToggleHandler, imageInferenceAdvancedFeaturesToggleHandler, watermarkAdvancedFeatureToggleHandler, videoInferenceSpeechInputToggleHandler, regionalPromptingRegionsToggleHandler, upscalerToggleHandler, imageUpscalerSettingsToggleHandler, videoUpscalerToggleHandler, audioInferenceToggleHandler, audioInferenceSpeechToggleHandler, audioSettingsToggleHandler, textInferenceSettingsToggleHandler, videoSettingsToggleHandler, videoInferenceSettingsTtsToggleHandler, videoInferenceSettingsActiveSpeakerDetectionToggleHandler, videoInferenceSettingsActiveSpeakerBoundingBoxesToggleHandler, videoInferenceSettingsSegmentsToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, xaiProviderSettingsToggleHandler, viduProviderSettingsToggleHandler, sourcefulProviderSettingsToggleHandler, sourcefulProviderSettingsFontsToggleHandler, threeDInferenceToggleHandler, threeDInferenceSettingsToggleHandler, threeDInferenceSettingsLatToggleHandler, threeDInferenceSettingsMeshClusterToggleHandler, ultralyticsProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, lightricksProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler, briaProviderSettingsToggleHandler, pixverseProviderSettingsToggleHandler, alibabaProviderSettingsToggleHandler, mireloProviderSettingsToggleHandler, googleProviderSettingsToggleHandler, syncProviderSettingsToggleHandler, syncSegmentToggleHandler, settingsToggleHandler, outpaintSettingsToggleHandler, safetyInputsToggleHandler, imageInferenceSettingsColorPaletteToggleHandler, imageInferenceSettingsMoodboardsToggleHandler, imageInferenceSettingsStructuredPromptToggleHandler, imageInferenceSettingsPromptEnhanceToggleHandler, imageInferenceSettingsScoringRubricToggleHandler, audioInputToggleHandler, speechInputToggleHandler, briaProviderMaskToggleHandler, wanAnimateAdvancedFeatureSettingsToggleHandler, videoAdvancedFeatureInputsToggleHandler, audioInferenceInputsToggleHandler, audioInferenceReferenceVoiceToggleHandler, audioInferenceSpeechVoicesToggleHandler, videoInputsFrameToggleHandler, referenceVideosToggleHandler } from "./utils.js";
import { promptEnhanceHandler, syncDimensionsNodeHandler, searchNodeHandler, APIKeyHandler, captionNodeHandler, saveTextHandler, mediaUUIDHandler, save3DFilepathHandler, videoTranscriptionHandler, videoOutputsHandler, handleCustomErrors, videoInferenceDimensionsHandler, videoModelSearchFilterHandler, threeDModelSearchFilterHandler, audioModelSearchFilterHandler, textModelSearchFilterHandler, vectorizeModelSearchFilterHandler, vectorizeToggleHandler, useParameterToggleHandler, imageInferenceToggleHandler, imageInferenceAdvancedFeaturesToggleHandler, watermarkAdvancedFeatureToggleHandler, videoInferenceSpeechInputToggleHandler, regionalPromptingRegionsToggleHandler, upscalerToggleHandler, imageUpscalerSettingsToggleHandler, videoUpscalerToggleHandler, audioInferenceToggleHandler, audioInferenceSpeechToggleHandler, audioSettingsToggleHandler, textInferenceSettingsToggleHandler, videoSettingsToggleHandler, videoInferenceSettingsTtsToggleHandler, videoInferenceSettingsActiveSpeakerDetectionToggleHandler, videoInferenceSettingsActiveSpeakerBoundingBoxesToggleHandler, videoInferenceSettingsSegmentsToggleHandler, acceleratorOptionsToggleHandler, bytedanceProviderSettingsToggleHandler, xaiProviderSettingsToggleHandler, viduProviderSettingsToggleHandler, sourcefulProviderSettingsToggleHandler, sourcefulProviderSettingsFontsToggleHandler, threeDInferenceToggleHandler, threeDInferenceSettingsToggleHandler, threeDInferenceSettingsLatToggleHandler, threeDInferenceSettingsMeshClusterToggleHandler, threeDInferenceSettingsDracoCompressionToggleHandler, ultralyticsProviderSettingsToggleHandler, openaiProviderSettingsToggleHandler, lightricksProviderSettingsToggleHandler, klingProviderSettingsToggleHandler, lumaProviderSettingsToggleHandler, briaProviderSettingsToggleHandler, pixverseProviderSettingsToggleHandler, alibabaProviderSettingsToggleHandler, mireloProviderSettingsToggleHandler, googleProviderSettingsToggleHandler, syncProviderSettingsToggleHandler, syncSegmentToggleHandler, settingsToggleHandler, outpaintSettingsToggleHandler, safetyInputsToggleHandler, imageInferenceSettingsColorPaletteToggleHandler, imageInferenceSettingsMoodboardsToggleHandler, imageInferenceSettingsStructuredPromptToggleHandler, imageInferenceSettingsPromptEnhanceToggleHandler, imageInferenceSettingsScoringRubricToggleHandler, audioInputToggleHandler, speechInputToggleHandler, briaProviderMaskToggleHandler, wanAnimateAdvancedFeatureSettingsToggleHandler, videoAdvancedFeatureInputsToggleHandler, audioInferenceInputsToggleHandler, audioInferenceReferenceVoiceToggleHandler, audioInferenceSpeechVoicesToggleHandler, videoInputsFrameToggleHandler, referenceVideosToggleHandler } from "./utils.js";
import { RUNWARE_NODE_TYPES, RUNWARE_NODE_PROPS, SEARCH_TERMS } from "./types.js";
const nodeInitList = [];
@@ -188,6 +188,8 @@ app.registerExtension({
threeDInferenceSettingsLatToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.THREEDINFERENCESETTINGSMESHCLUSTER) {
threeDInferenceSettingsMeshClusterToggleHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.THREEDINFERENCESETTINGSDRACOCOMPRESSION) {
threeDInferenceSettingsDracoCompressionToggleHandler(node);
}
if(crNodeProps.colorModeOnly === true) return;
@@ -195,6 +197,8 @@ app.registerExtension({
if(nodeClass === RUNWARE_NODE_TYPES.VIDEOMODELSEARCH) {
videoModelSearchFilterHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.THREEDMODELSEARCH) {
threeDModelSearchFilterHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.AUDIOMODELSEARCH) {
audioModelSearchFilterHandler(node);
} else if(nodeClass === RUNWARE_NODE_TYPES.TEXTMODELSEARCH) {
clientlibs/types.js
+10
View File
@@ -170,12 +170,14 @@ const RUNWARE_NODE_TYPES = {
SAVETEXT: "Runware Save Text",
VIDEOINFERENCEOUTPUTS: "Runware Video Inference Outputs",
THREEDINFERENCE: "Runware 3D Inference",
THREEDMODELSEARCH: "Runware 3D Model Search",
THREEDINFERENCEINPUTS: "Runware 3D Inference Inputs",
THREEDINFERENCESETTINGS: "Runware 3D Inference Settings",
THREEDINFERENCESETTINGSSPARSESTRUCTURE: "Runware 3D Inference Settings Sparse Structure",
THREEDINFERENCESETTINGSSHAPESLAT: "Runware 3D Inference Settings Shape Slat",
THREEDINFERENCESETTINGSTEXSLAT: "Runware 3D Inference Settings Tex Slat",
THREEDINFERENCESETTINGSMESHCLUSTER: "Runware 3D Inference Settings Mesh Cluster",
THREEDINFERENCESETTINGSDRACOCOMPRESSION: "Runware 3D Inference Settings Draco Compression",
SAVE3D: "Runware Save 3D",
LOADMESH: "Runware Load Mesh",
WATERMARKADVANCEDFEATURE: "Runware Watermark Advanced Feature",
@@ -681,6 +683,10 @@ const RUNWARE_NODE_PROPS = {
bgColor: DEFAULT_BGCOLOR,
promptEnhancer: true,
},
[RUNWARE_NODE_TYPES.THREEDMODELSEARCH]: {
bgColor: DEFAULT_BGCOLOR,
liveSearch: false,
},
[RUNWARE_NODE_TYPES.THREEDINFERENCEINPUTS]: {
bgColor: DEFAULT_BGCOLOR,
colorModeOnly: true,
@@ -705,6 +711,10 @@ const RUNWARE_NODE_PROPS = {
bgColor: DEFAULT_BGCOLOR,
colorModeOnly: true,
},
[RUNWARE_NODE_TYPES.THREEDINFERENCESETTINGSDRACOCOMPRESSION]: {
bgColor: DEFAULT_BGCOLOR,
colorModeOnly: true,
},
[RUNWARE_NODE_TYPES.SAVE3D]: {
bgColor: DEFAULT_BGCOLOR,
colorModeOnly: true,
clientlibs/utils.js
+79
View File
@@ -2324,6 +2324,28 @@ function threeDInferenceSettingsMeshClusterToggleHandler(node) {
});
}
function threeDInferenceSettingsDracoCompressionToggleHandler(node) {
const pairs = [
["useEnabled", "enabled"],
["useLevel", "level"],
["useQuantizationPosition", "quantizationPosition"],
["useQuantizationNormal", "quantizationNormal"],
["useQuantizationTexCoord", "quantizationTexCoord"],
];
pairs.forEach(([useName, paramName]) => {
const useW = node.widgets.find(w => w.name === useName);
const paramW = node.widgets.find(w => w.name === paramName);
if (!useW || !paramW) return;
function toggleEnabled() {
const enabled = useW.value === true;
toggleWidgetEnabled(paramW, enabled, node);
node.setDirtyCanvas(true);
}
appendWidgetCB(useW, () => setTimeout(toggleEnabled, 50));
setTimeout(toggleEnabled, 100);
});
}
function openaiProviderSettingsToggleHandler(openaiNode) {
// Find all "use" parameter widgets for OpenAI Provider Settings (these are COMBO widgets)
const useBackgroundWidget = openaiNode.widgets.find(w => w.name === "useBackground");
@@ -3416,6 +3438,61 @@ function textModelSearchFilterHandler(textModelSearchNode) {
filterModelList();
}
function threeDModelSearchFilterHandler(threeDModelSearchNode) {
const modelArchWidget = threeDModelSearchNode.widgets.find(w => w.name === "Model Architecture");
const threeDListWidget = threeDModelSearchNode.widgets.find(w => w.name === "ThreeDList");
if (!modelArchWidget || !threeDListWidget) return;
const THREED_MODELS = {
"Tencent": [
"tencent:hunyuan-3d@3.1-pro (Tencent Hunyuan 3D 3.1 Pro)",
"tencent:hunyuan-3d@3.1-rapid (Tencent Hunyuan 3D 3.1 Rapid)",
],
"Meta": [
"meta:sam@3d (Meta SAM 3D)",
],
"Microsoft": [
"microsoft:trellis-2@4b (TRELLIS.2)",
],
"Tripo": [
"tripo:v3.1@0 (Tripo 3D v3.1)",
],
"Hyper3D": [
"hyper3d:rodin@gen-1 (Rodin Gen-1)",
"hyper3d:rodin@gen-2 (Rodin Gen-2)",
],
"Meshy": [
"meshy:meshy@6 (Meshy 6)",
],
};
function filterModelList() {
const selectedArch = modelArchWidget.value;
let filteredModels = [];
if (selectedArch === "All") {
Object.values(THREED_MODELS).forEach(models => filteredModels.push(...models));
} else if (THREED_MODELS[selectedArch]) {
filteredModels = THREED_MODELS[selectedArch];
}
if (filteredModels.length > 0) {
const currentValue = threeDListWidget.value;
threeDListWidget.options.values = filteredModels;
if (!filteredModels.includes(currentValue)) {
threeDListWidget.value = filteredModels[0];
}
threeDModelSearchNode.setDirtyCanvas(true);
}
}
appendWidgetCB(modelArchWidget, filterModelList);
filterModelList();
}
function audioModelSearchFilterHandler(audioModelSearchNode) {
const modelProviderWidget = audioModelSearchNode.widgets.find(w => w.name === "Model Provider");
const audioListWidget = audioModelSearchNode.widgets.find(w => w.name === "AudioList");
@@ -5208,6 +5285,7 @@ export {
APIKeyHandler,
videoInferenceDimensionsHandler,
videoModelSearchFilterHandler,
threeDModelSearchFilterHandler,
audioModelSearchFilterHandler,
textModelSearchFilterHandler,
vectorizeModelSearchFilterHandler,
@@ -5240,6 +5318,7 @@ export {
threeDInferenceSettingsToggleHandler,
threeDInferenceSettingsLatToggleHandler,
threeDInferenceSettingsMeshClusterToggleHandler,
threeDInferenceSettingsDracoCompressionToggleHandler,
ultralyticsProviderSettingsToggleHandler,
openaiProviderSettingsToggleHandler,
lightricksProviderSettingsToggleHandler,
modules/threeDInference.py
+4 -20
View File
@@ -5,18 +5,6 @@ import comfy.model_management
class threeDInference:
"""Runware 3D Inference node for generating 3D models from images"""
# Available 3D models
THREED_MODELS = {
"Tencent Hunyuan 3D 3.1 Pro": "tencent:hunyuan-3d@3.1-pro",
"Tencent Hunyuan 3D 3.1 Rapid": "tencent:hunyuan-3d@3.1-rapid",
"Meta SAM 3D": "meta:sam@3d",
"TRELLIS.2": "microsoft:trellis-2@4b",
"Tripo 3D v3.1": "tripo:v3.1@0",
"Rodin Gen-1": "hyper3d:rodin@gen-1",
"Rodin Gen-2": "hyper3d:rodin@gen-2",
"Meshy 6": "meshy:meshy@6",
}
# Output formats
OUTPUT_FORMATS = ["GLB", "FBX", "PLY", "OBJ"]
@@ -24,9 +12,8 @@ class threeDInference:
def INPUT_TYPES(cls):
return {
"required": {
"Model": (list(cls.THREED_MODELS.keys()), {
"tooltip": "Select the 3D generation model to use.",
"default": "Meta SAM 3D",
"Model": ("RUNWARE3DMODEL", {
"tooltip": "Connect a Runware 3D Model from Runware 3D Model Search node.",
}),
"positivePrompt": ("STRING", {
"multiline": True,
@@ -66,7 +53,7 @@ class threeDInference:
"step": 1,
}),
"inputs": ("RUNWARE3DINFERENCEINPUTS", {
"tooltip": "Connect Runware 3D Inference Inputs for image, mask, meshFile, images (array from images_1…4), etc.",
"tooltip": "Connect Runware 3D Inference Inputs for image, mask, meshFile, images (array from Images 1…8), etc.",
}),
"settings": ("RUNWARE3DINFERENCESETTINGS", {
"tooltip": "Connect Runware 3D Inference Settings for textureSize, decimationTarget, remesh, resolution, imageAutoFix, faceLimit, texture, pbr, quad, Tripo options, sparseStructure, shapeSlat, texSlat, etc.",
@@ -82,7 +69,7 @@ class threeDInference:
def generate3D(self, **kwargs):
"""Generate 3D model from inputs"""
modelName = kwargs.get("Model", "Meta SAM 3D")
model = kwargs.get("Model", "meta:sam@3d")
positivePrompt = kwargs.get("positivePrompt", "")
negativePrompt = kwargs.get("negativePrompt", "")
seed = kwargs.get("seed", 1)
@@ -93,9 +80,6 @@ class threeDInference:
inputs = kwargs.get("inputs", None)
settings = kwargs.get("settings", None)
# Get model AIR code
model = self.THREED_MODELS.get(modelName, "meta:sam@3d")
# Build generation config
genConfig = [
{
modules/threeDInferenceInputs.py
+3 -3
View File
@@ -21,7 +21,7 @@ class threeDInferenceInputs:
}),
}
for i in range(1, _IMAGE_SLOTS + 1):
optional[f"images_{i}"] = ("IMAGE", {
optional[f"Images {i}"] = ("IMAGE", {
"tooltip": f"Multi-view image slot {i}; merged in order into inputs.images as a list of data URIs.",
})
return {
@@ -30,7 +30,7 @@ class threeDInferenceInputs:
}
DESCRIPTION = (
"Configure custom inputs for Runware 3D Inference: image, mask, meshFile, and images_1…images_8 "
"Configure custom inputs for Runware 3D Inference: image, mask, meshFile, and Images 1…Images 8 "
"(merged into inputs.images as an array)."
)
FUNCTION = "createInputs"
@@ -57,7 +57,7 @@ class threeDInferenceInputs:
images_list: list = []
for i in range(1, _IMAGE_SLOTS + 1):
slot = kwargs.get(f"images_{i}")
slot = kwargs.get(f"Images {i}")
if slot is not None:
images_list.append(rwUtils.convertTensor2IMG(slot))
if images_list:
modules/threeDInferenceSettings.py
+9 -2
View File
@@ -3,7 +3,7 @@ Runware 3D Inference Settings Node
Main settings for 3D inference: textureSize, decimationTarget, remesh, remeshBand, remeshProject,
resolution, imageAutoFix, faceLimit, texture, pbr, textureFormat, alphaMode, Tripo mesh/texture options, material/quality,
polyCount, taPose, boundingBox, meshMode, addons, hdTexture, and sub-configs
sparseStructure, shapeSlat, texSlat, meshCluster from connected nodes.
sparseStructure, shapeSlat, texSlat, meshCluster, dracoCompression from connected nodes.
"""
from typing import Dict, Any
@@ -466,6 +466,9 @@ class Runware3DInferenceSettings:
"tooltip": "Material alpha mode. Only used when 'Use Alpha Mode' is enabled.",
"default": "OPAQUE",
}),
"dracoCompression": ("RUNWARE3DINFERENCESETTINGSLAT", {
"tooltip": "Connect Runware 3D Inference Settings Draco Compression node.",
}),
}
}
@@ -478,7 +481,7 @@ class Runware3DInferenceSettings:
"resolution, imageAutoFix, faceLimit, texture, pbr, textureSeed, textureAlignment, textureQuality, "
"textureFormat, alphaMode, autoSize, orientation, quad, compress, "
"smartLowPoly, generateParts, exportUv, geometryQuality, originalAlpha, material, quality, polyCount, "
"taPose, boundingBox, meshMode, addons, hdTexture, lat configs, and meshCluster."
"taPose, boundingBox, meshMode, addons, hdTexture, lat configs, meshCluster, and dracoCompression."
)
def create(self, **kwargs) -> tuple[Dict[str, Any]]:
@@ -610,6 +613,10 @@ class Runware3DInferenceSettings:
if mesh_cluster is not None and isinstance(mesh_cluster, dict) and len(mesh_cluster) > 0:
settings["meshCluster"] = mesh_cluster
draco_compression = kwargs.get("dracoCompression", None)
if draco_compression is not None and isinstance(draco_compression, dict) and len(draco_compression) > 0:
settings["dracoCompression"] = draco_compression
return (settings,)
modules/threeDInferenceSettingsDracoCompression.py
+109
View File
@@ -0,0 +1,109 @@
"""
Runware 3D Inference Settings Draco Compression Node
Outputs dracoCompression config: enabled, level, quantizationPosition,
quantizationNormal, quantizationTexCoord
"""
from typing import Dict, Any
class Runware3DInferenceSettingsDracoCompression:
"""Runware 3D Inference Settings Draco Compression Node"""
@classmethod
def INPUT_TYPES(cls):
return {
"required": {},
"optional": {
"useEnabled": ("BOOLEAN", {
"tooltip": "Enable to include enabled in dracoCompression",
"default": False,
}),
"enabled": ("BOOLEAN", {
"tooltip": (
"If true, applies KHR_draco_mesh_compression to the output GLB "
"(5-10x geometry size reduction with no triangle loss). "
"Only used when 'Use Enabled' is on."
),
"default": False,
"label_on": "true",
"label_off": "false",
}),
"useLevel": ("BOOLEAN", {
"tooltip": "Enable to include level in dracoCompression",
"default": False,
}),
"level": ("INT", {
"tooltip": "Draco compression level (0-10). Higher = better compression but slower encoding. Only used when enabled.",
"default": 7,
"min": 0,
"max": 10,
"step": 1,
}),
"useQuantizationPosition": ("BOOLEAN", {
"tooltip": "Enable to include quantizationPosition in dracoCompression",
"default": False,
}),
"quantizationPosition": ("INT", {
"tooltip": "Quantization bits for vertex positions (8-30). Higher = better quality, larger file. Only used when enabled.",
"default": 16,
"min": 8,
"max": 30,
"step": 1,
}),
"useQuantizationNormal": ("BOOLEAN", {
"tooltip": "Enable to include quantizationNormal in dracoCompression",
"default": False,
}),
"quantizationNormal": ("INT", {
"tooltip": "Quantization bits for vertex normals (8-30). Higher = smoother shading, larger file. Only used when enabled.",
"default": 14,
"min": 8,
"max": 30,
"step": 1,
}),
"useQuantizationTexCoord": ("BOOLEAN", {
"tooltip": "Enable to include quantizationTexCoord in dracoCompression",
"default": False,
}),
"quantizationTexCoord": ("INT", {
"tooltip": "Quantization bits for texture coordinates (8-30). Higher = sharper textures at UV seams. Only used when enabled.",
"default": 14,
"min": 8,
"max": 30,
"step": 1,
}),
}
}
RETURN_TYPES = ("RUNWARE3DINFERENCESETTINGSLAT",)
RETURN_NAMES = ("Draco Compression",)
FUNCTION = "create"
CATEGORY = "Runware"
DESCRIPTION = (
"Configure settings.dracoCompression for Runware 3D Inference "
"(enabled, level, quantizationPosition, quantizationNormal, quantizationTexCoord). "
"Connect to Runware 3D Inference Settings."
)
def create(self, **kwargs) -> tuple[Dict[str, Any]]:
out: Dict[str, Any] = {}
if kwargs.get("useEnabled", False):
out["enabled"] = bool(kwargs.get("enabled", False))
if kwargs.get("useLevel", False):
out["level"] = int(kwargs.get("level", 7))
if kwargs.get("useQuantizationPosition", False):
out["quantizationPosition"] = int(kwargs.get("quantizationPosition", 16))
if kwargs.get("useQuantizationNormal", False):
out["quantizationNormal"] = int(kwargs.get("quantizationNormal", 14))
if kwargs.get("useQuantizationTexCoord", False):
out["quantizationTexCoord"] = int(kwargs.get("quantizationTexCoord", 14))
return (out,)
NODE_CLASS_MAPPINGS = {
"Runware3DInferenceSettingsDracoCompression": Runware3DInferenceSettingsDracoCompression,
}
NODE_DISPLAY_NAME_MAPPINGS = {
"Runware3DInferenceSettingsDracoCompression": "Runware 3D Inference Settings Draco Compression",
}
modules/threeDModelSearch.py
+91
View File
@@ -0,0 +1,91 @@
class Runware3DModelSearch:
"""3D Model Search node for searching and selecting 3D generation models"""
THREED_MODELS = {
"Tencent": [
"tencent:hunyuan-3d@3.1-pro (Tencent Hunyuan 3D 3.1 Pro)",
"tencent:hunyuan-3d@3.1-rapid (Tencent Hunyuan 3D 3.1 Rapid)",
],
"Meta": [
"meta:sam@3d (Meta SAM 3D)",
],
"Microsoft": [
"microsoft:trellis-2@4b (TRELLIS.2)",
],
"Tripo": [
"tripo:v3.1@0 (Tripo 3D v3.1)",
],
"Hyper3D": [
"hyper3d:rodin@gen-1 (Rodin Gen-1)",
"hyper3d:rodin@gen-2 (Rodin Gen-2)",
],
"Meshy": [
"meshy:meshy@6 (Meshy 6)",
],
}
MODEL_ARCHITECTURES = [
"All",
"Tencent",
"Meta",
"Microsoft",
"Tripo",
"Hyper3D",
"Meshy",
]
@classmethod
def INPUT_TYPES(cls):
allModels = cls._getAllModels()
defaultModel = "meta:sam@3d (Meta SAM 3D)"
return {
"required": {
"Model Search": ("STRING", {
"tooltip": "Search for a specific 3D model by name or AIR code (e.g. hunyuan, trellis, tripo).",
}),
"Model Architecture": (cls.MODEL_ARCHITECTURES, {
"tooltip": "Choose 3D model architecture to filter results.",
"default": "Meta",
}),
"ThreeDList": (allModels, {
"tooltip": "3D model results will show up here so you can choose from.",
"default": defaultModel,
}),
"Use Search Value": ("BOOLEAN", {
"tooltip": "When enabled, the value in the search input is used instead of the selected list entry.\n\nUseful if you prefer to set the model manually.",
"default": False,
"label_on": "Enabled",
"label_off": "Disabled",
}),
},
}
@classmethod
def _getAllModels(cls):
allModels = []
for models in cls.THREED_MODELS.values():
allModels.extend(models)
return allModels
DESCRIPTION = "Search and connect 3D models to Runware 3D Inference nodes in ComfyUI."
FUNCTION = "threeDModelSearch"
RETURN_TYPES = ("RUNWARE3DMODEL",)
RETURN_NAMES = ("Runware 3D Model",)
CATEGORY = "Runware"
@classmethod
def VALIDATE_INPUTS(cls, ThreeDList):
return True
def threeDModelSearch(self, **kwargs):
enableSearchValue = kwargs.get("Use Search Value", False)
searchInput = kwargs.get("Model Search")
if enableSearchValue:
modelAirCode = searchInput
else:
currentModel = kwargs.get("ThreeDList")
modelAirCode = currentModel.split(" (")[0]
return (modelAirCode,)
workflows/Runware_3D_Inference_Trellis_2_Draco_Compression.json
+1
View File
File diff suppressed because one or more lines are too long