[Bug] ksampler error #535

Checklist

• 1. I have searched for related issues and FAQs (https://github.com/mit-han-lab/nunchaku/blob/main/docs/faq.md) but was unable to find a solution.
• 2. The issue persists in the latest version.
• 3. Please note that without environment information and a minimal reproducible example, it will be difficult for us to reproduce and address the issue, which may delay our response.
• 4. If your report is a question rather than a bug, please submit it as a discussion at https://github.com/mit-han-lab/ComfyUI-nunchaku/discussions/new/choose. Otherwise, this issue will be closed.
• 5. I will do my best to describe the issue in English.

Describe the Bug

when processing ksampler, it shows an error: view size is not compatible with input tensor's size and stride (at least one dimension spans across two contiguous subspaces). Use .reshape(...) instead.

Environment

nunchaku: v1.0.0
comfyui-nunchaku: v1.0.0
error report.txt

Reproduction Steps

workflow.json