AnswerDotAI / AnswerDotAI/fsdp_qlora

Torch Compile?

Open
#34 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Jupyter Notebook
Stars
1.6k
Forks
201
PR merge metrics
No merged PRs in 30d

Description

Thanks for such wonderful work!
I see you comment out this line:

https://github.com/AnswerDotAI/fsdp_qlora/blob/d7818ec86d17f37db4beef36f80870cbcac37957/train.py#L722

May I ask what is the rationale behind it? Is fsdp_qlora compatible with torch compile?

Contributor guide

No contributing guide indexed for this repository

Research direction

Start with train.py at line 722 and inspect the commented-out line in the referenced commit. Read the surrounding training setup and determine whether torch.compile is intended to work with the FSDP and QLoRA configuration; the issue is done when the rationale and compatibility status are established and clearly documented.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
distributed-systems, machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
15/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.