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

Merge pull request #169 from filipstrand/remove-torchvision

Browse Source 
Remove TorchVision Dependency
This commit is contained in:
Filip Strand 2025-04-24 15:50:34 +02:00 committed by GitHub
commit 8743d1d4ed
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
pyproject.toml
View File

@ -29,7 +29,6 @@ dependencies = [
"tokenizers>=0.20.3; python_version>='3.13'", # transformers -> tokenizers
"toml>=0.10.2,<1.0",
"torch>=2.3.1,<3.0; python_version<'3.13'",
"torchvision>=0.18.1,<0.22.0",
# torch dev builds: pip install --pre --index-url https://download.pytorch.org/whl/nightly
"torch>=2.6.0.dev20241106; python_version>='3.13'",
"tqdm>=4.66.5,<5.0",

17
src/mflux/post_processing/image_util.py
View File

@ -8,7 +8,6 @@ import numpy as np
import piexif
import PIL.Image
import PIL.ImageDraw
import torchvision.transforms.functional as F
from mflux.config.runtime_config import RuntimeConfig
from mflux.post_processing.generated_image import GeneratedImage
@ -302,6 +301,16 @@ class ImageUtil:
std: list = [0.5, 0.5, 0.5],
resample: int = PIL.Image.LANCZOS,
) -> mx.array:
image_np = F.to_tensor(image)
tensor = F.normalize(image_np, mean, std, False)
return mx.array(tensor)
# Convert PIL image to numpy array and normalize to [0, 1]
image_np = np.array(image).astype(np.float32) / 255.0
# Convert from HWC to CHW format
image_np = image_np.transpose(2, 0, 1)
# Normalize using specified mean and std
mean_np = np.array(mean).reshape(-1, 1, 1)
std_np = np.array(std).reshape(-1, 1, 1)
image_np = (image_np - mean_np) / std_np
# Convert to MLX array
return mx.array(image_np)