monster/Qwen-Image-Layered-MRP-MLX
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix sigmas shift

Browse Source
This commit is contained in:
Fabio 2024-08-18 17:15:58 +02:00
parent 571608b4dd
commit 6977e92061
2 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/flux_1_schnell/config/config.py
View File

@ -34,5 +34,5 @@ class Config:
x1 = 256 x1 = 256
m = (1.15 - y1) / (4096 - x1) m = (1.15 - y1) / (4096 - x1)
b = y1 - m * x1 b = y1 - m * x1
mu = m + b mu = m * self.width * self.height / 256 + b
self.sigmas = mx.exp(mu) / (mx.exp(mu) + (1 / self.sigmas - 1)) self.sigmas = mx.exp(mu) / (mx.exp(mu) + (1 / self.sigmas - 1))

3
src/flux_1_schnell/flux.py
View File

@ -20,8 +20,7 @@ class Flux1:
def __init__(self, repo_id: str): def __init__(self, repo_id: str):
self.is_dev = "FLUX.1-dev" in repo_id self.is_dev = "FLUX.1-dev" in repo_id
# max_t5_length = 512 if self.is_dev else 256 max_t5_length = 512 if self.is_dev else 256
max_t5_length = 256
tokenizers = TokenizerHandler.load_from_disk_or_huggingface(repo_id, max_t5_length) tokenizers = TokenizerHandler.load_from_disk_or_huggingface(repo_id, max_t5_length)
self.t5_tokenizer = TokenizerT5(tokenizers.t5, max_length=max_t5_length) self.t5_tokenizer = TokenizerT5(tokenizers.t5, max_length=max_t5_length)
self.clip_tokenizer = TokenizerCLIP(tokenizers.clip) self.clip_tokenizer = TokenizerCLIP(tokenizers.clip)