kohya-ss / kohya-ss/sd-scripts

pytorch 2.4 is too old.

Open
#2,119 0 comments 0 reactions 0 assignees View on GitHub
enhancement
Dominant language
Python
Stars
7.2k
Forks
1.2k
Avg merge
11m
Merged PRs (30d)
2

Description

The sd3 branch recommends using pytorch 2.4, but this is too old for new hardware (such as 5090 or MI300). I tried installing and running with 2.7, but got lots of errors when I tried to train a Lora. It looks like some effort is required to upgrade all the dependencies to a newer version of pytorch.

Contributor guide

No contributing guide indexed for this repository

Research direction

Start by reviewing the sd3 branch dependency setup and reproducing LoRA training with PyTorch 2.7. Identify the dependency and compatibility errors that occur on newer hardware such as the 5090 or MI300. Done means the sd3 branch supports a newer PyTorch version and LoRA training runs without those errors.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.