Skip to content

Simply device allocation in finetuning notebook#1163

Merged
anwai98 merged 2 commits intodevfrom
simplify-device-allocation
Feb 11, 2026
Merged

Simply device allocation in finetuning notebook#1163
anwai98 merged 2 commits intodevfrom
simplify-device-allocation

Conversation

@anwai98
Copy link
Collaborator

@anwai98 anwai98 commented Feb 11, 2026

Easy peasy PR. Will merge this once the tests pass!

@anwai98 anwai98 self-assigned this Feb 11, 2026
@review-notebook-app
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@anwai98 anwai98 merged commit f2f34f5 into dev Feb 11, 2026
5 of 6 checks passed
@anwai98 anwai98 deleted the simplify-device-allocation branch February 11, 2026 15:35
anwai98 added a commit that referenced this pull request Feb 11, 2026
* Fix issue in batched inference

* Feaure: Added Apple silicon support in nb finetuning. (#1161)

Checks if Cuda or MPS (Apple Silicon) is available. Otherwise defaults to GPU.

* Simply device allocation in finetuning notebook (#1163)

* Simply device allocation in finetuning notebook

* Update finetuning python scripts

---------

Co-authored-by: Constantin Pape <constantin.pape@informatik.uni-goettingen.de>
Co-authored-by: Hendrik Windel <15370203+HendrikLamar@users.noreply.github.com>
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

Comments