| 21 | |
| 22 | |
| 23 | class StableFast3DLoader: |
| 24 | CATEGORY = SF3D_CATEGORY |
| 25 | FUNCTION = "load" |
| 26 | RETURN_NAMES = ("sf3d_model",) |
| 27 | RETURN_TYPES = ("SF3D_MODEL",) |
| 28 | |
| 29 | @classmethod |
| 30 | def INPUT_TYPES(cls): |
| 31 | return {"required": {}} |
| 32 | |
| 33 | def load(self): |
| 34 | device = comfy.model_management.get_torch_device() |
| 35 | model = SF3D.from_pretrained( |
| 36 | SF3D_MODEL_NAME, |
| 37 | config_name="config.yaml", |
| 38 | weight_name="model.safetensors", |
| 39 | ) |
| 40 | model.to(device) |
| 41 | model.eval() |
| 42 | |
| 43 | return (model,) |
| 44 | |
| 45 | |
| 46 | class StableFast3DPreview: |
nothing calls this directly
no outgoing calls
no test coverage detected