github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

Fix incorrect device argument initialization for LoRA training (#1231)

Browse Source 
Co-authored-by: Kyle Herndon <kyle@nod-labs.com>
Co-authored-by: powderluv <powderluv@users.noreply.github.com>
This commit is contained in:
Kyle Herndon
2023-03-21 19:07:18 -07:00
committed by GitHub
parent 7dbbb1726a
commit c23fcf3748
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
apps/stable_diffusion/scripts/train_lora_word.py
View File

@@ -227,7 +227,7 @@ def lora_train(
args.max_length = max_length
args.height = height
args.width = width
args.device = device
args.device = device.split("=>", 1)[1].strip()
# Load the Stable Diffusion model
text_encoder = CLIPTextModel.from_pretrained(