Skip to content

fix:guard torch.compile in get_smol so first index doesnt crash in windows#30

Open
Qayad-Ali wants to merge 1 commit into
owaismohammad:mainfrom
Qayad-Ali:fix/torch-compile-windows
Open

fix:guard torch.compile in get_smol so first index doesnt crash in windows#30
Qayad-Ali wants to merge 1 commit into
owaismohammad:mainfrom
Qayad-Ali:fix/torch-compile-windows

Conversation

@Qayad-Ali

Copy link
Copy Markdown

Hey ,was using gurrt on my machine and came across a bug . it didnt work on my machine bcs I didn't have a C++ compiler installed.Thought i should contribute for this.

What/Issue

ModelManager.get_smol() always calls torch.compile(). On Windows, this can fail because the CPU backend requires MSVC (cl.exe) and CUDA compilation depends on Triton, which isn't supported on Windows.

Impact

Following the README quick-start (/index smolvlm) causes indexing to crash on Windows before any frames are captioned. Reproduced with PyTorch 2.12.1 (CPU).

Fix

Only compile on CUDA-enabled non-Windows systems, and fall back to the eager model if compilation fails. No behavior change for supported Linux/CUDA setups.

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

Successfully merging this pull request may close these issues.

1 participant