jaidaken
f09734b0ee
Some checks failed
Python Linting / Run Ruff (push) Has been cancelled
Python Linting / Run Pylint (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Has been cancelled
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Has been cancelled
Execution Tests / test (macos-latest) (push) Has been cancelled
Execution Tests / test (ubuntu-latest) (push) Has been cancelled
Execution Tests / test (windows-latest) (push) Has been cancelled
Test server launches without errors / test (push) Has been cancelled
Unit Tests / test (macos-latest) (push) Has been cancelled
Unit Tests / test (ubuntu-latest) (push) Has been cancelled
Unit Tests / test (windows-2022) (push) Has been cancelled
Includes 30 custom nodes committed directly, 7 Civitai-exclusive loras stored via Git LFS, and a setup script that installs all dependencies and downloads HuggingFace-hosted models on vast.ai. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
38 lines
1.1 KiB
Python
38 lines
1.1 KiB
Python
import torch
|
|
from comfy import model_management
|
|
|
|
def string_to_dtype(s="none", mode=None):
|
|
s = s.lower().strip()
|
|
if s in ["default", "as-is"]:
|
|
return None
|
|
elif s in ["auto", "auto (comfy)"]:
|
|
if mode == "vae":
|
|
return model_management.vae_device()
|
|
elif mode == "text_encoder":
|
|
return model_management.text_encoder_dtype()
|
|
elif mode == "unet":
|
|
return model_management.unet_dtype()
|
|
else:
|
|
raise NotImplementedError(f"Unknown dtype mode '{mode}'")
|
|
elif s in ["none", "auto (hf)", "auto (hf/bnb)"]:
|
|
return None
|
|
elif s in ["fp32", "float32", "float"]:
|
|
return torch.float32
|
|
elif s in ["bf16", "bfloat16"]:
|
|
return torch.bfloat16
|
|
elif s in ["fp16", "float16", "half"]:
|
|
return torch.float16
|
|
elif "fp8" in s or "float8" in s:
|
|
if "e5m2" in s:
|
|
return torch.float8_e5m2
|
|
elif "e4m3" in s:
|
|
return torch.float8_e4m3fn
|
|
else:
|
|
raise NotImplementedError(f"Unknown 8bit dtype '{s}'")
|
|
elif "bnb" in s:
|
|
assert s in ["bnb8bit", "bnb4bit"], f"Unknown bnb mode '{s}'"
|
|
return s
|
|
elif s is None:
|
|
return None
|
|
else:
|
|
raise NotImplementedError(f"Unknown dtype '{s}'") |