theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-25 10:48:11 +00:00
Code Issues Packages Projects Releases Wiki Activity

Flux lora (#205)

Browse Source
This commit is contained in:
Zhongjie Duan
2024-09-12 16:49:30 +08:00
committed by GitHub
parent a8cb4a21d1
commit c21ed1e478
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
diffsynth/pipelines/sdxl_image.py
View File

@@ -9,6 +9,7 @@ from .dancer import lets_dance_xl
from typing import List from typing import List
import torch import torch
from tqdm import tqdm from tqdm import tqdm
from einops import repeat
@@ -103,7 +104,8 @@ class SDXLImagePipeline(BasePipeline):
def prepare_extra_input(self, latents=None): def prepare_extra_input(self, latents=None):
height, width = latents.shape[2] * 8, latents.shape[3] * 8 height, width = latents.shape[2] * 8, latents.shape[3] * 8
return {"add_time_id": torch.tensor([height, width, 0, 0, height, width], device=self.device)} add_time_id = torch.tensor([height, width, 0, 0, height, width], device=self.device).repeat(latents.shape[0])
return {"add_time_id": add_time_id}
@torch.no_grad() @torch.no_grad()

2
diffsynth/prompters/flux_prompter.py
View File

@@ -49,8 +49,6 @@ class FluxPrompter(BasePrompter):
truncation=True, truncation=True,
).input_ids.to(device) ).input_ids.to(device)
prompt_emb = text_encoder(input_ids) prompt_emb = text_encoder(input_ids)
prompt_emb = prompt_emb.reshape((1, prompt_emb.shape[0]*prompt_emb.shape[1], -1))
return prompt_emb return prompt_emb