Skip to content

For AMD (Directml) users #15

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
MythicalChu opened this issue Feb 5, 2024 · 1 comment
Open

For AMD (Directml) users #15

MythicalChu opened this issue Feb 5, 2024 · 1 comment

Comments

@MythicalChu
Copy link

MythicalChu commented Feb 5, 2024

Clone the repository and remove torch from requirements.txt (assuming you already have torch-directml installed) before pip installing it.
Delete or comment-out the line with torch.cuda.ipc_collect() from nodes.py on custom_nodes\ComfyUI-Marigold folder.

On the workflow, on the node:
reduction_method: mean, works on fp16
reduction_method: median, errors on fp16

scheduler: DEISMultistepScheduler, is the only one that doesn't freeze during the denoising step of the node.

That's how it works for me with my Rx580 (8Gb), on Windows. I only do 1 n_repeat_batch_size at 768 resolution or 2 at half that resolution, but maybe I can go higher. Half the recommended resolution is fast to compute and still gives better results than Midas, imo.

@KillyTheNetTerminal
Copy link

please can you help me? Rx580 here to, so I triying to install this custom node but Comfy just "Import Failed" I try manually and it's the same. I'm looking for torch.cudaipc_collect() too but I can't find it inside nodes.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants