theluyuan/DiffSynth-Studio
1
0
Fork 0
mirror of https://github.com/modelscope/DiffSynth-Studio.git synced 2026-03-19 06:32:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

update value controller

Browse Source
This commit is contained in:
Artiprocher
2025-07-21 16:30:06 +08:00
parent 1384de0353
commit 22705a44b4
4 changed files with 18 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
diffsynth/models/flux_value_control.py
View File

@@ -18,7 +18,7 @@ class MultiValueEncoder(torch.nn.Module):
class SingleValueEncoder(torch.nn.Module):
def __init__(self, dim_in=256, dim_out=3072, prefer_len=32, computation_device=None):
def __init__(self, dim_in=256, dim_out=4096, prefer_len=32, computation_device=None):
super().__init__()
self.prefer_len = prefer_len
self.prefer_proj = TemporalTimesteps(num_channels=dim_in, flip_sin_to_cos=True, downscale_freq_shift=0, computation_device=computation_device)