Comfy-Org / Comfy-Org/ComfyUI

FLUX Training lora error

Open
#7,141 1 comment 0 reactions 0 assignees View on GitHub
Potential Bug
Dominant language
Python
Stars
133k
Forks
15.7k
Avg merge
1d 6h
Merged PRs (30d)
155

Description

### Expected Behavior

My computer with 16GB VRAM,

### Actual Behavior

size mismatch for img_in.weight: copying a param with shape torch.Size([3072, 384]) from checkpoint, the shape in current model is torch.Size([3072, 64]).

### Steps to Reproduce

size mismatch for img_in.weight: copying a param with shape torch.Size([3072, 384]) from checkpoint, the shape in current model is torch.Size([3072, 64]).

### Debug Logs

```powershell
![Image](https://github.com/user-attachments/assets/d8d6e04c-641a-466f-a91f-24be27f3333f)

![Image](https://github.com/user-attachments/assets/40d96b50-2e57-4dab-9a9e-da379ae9235c)
```

### Other

_No response_

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.