Flux Schnell ist ein leistungsstarkes Bildgenerierungsmodell, das hochwertige Bilder aus Textvorgaben erstellen kann. In diesem Artikel zeigen wir Ihnen, wie Sie Flux Schnell lokal auf einem MacBook Pro mit einem M3 Max Chip einrichten und ausführen können. Wir werden die Anforderungen, Installationsvorgänge und ein Schritt-für-Schritt-Tutorial zur Erzeugung Ihres ersten Bildes behandeln.
Bevor wir beginnen, stellen Sie sicher, dass Ihr System die folgenden Anforderungen erfüllt:
Zerlegen wir den Installationsvorgang in überschaubare Schritte:
Wir installieren zunächst Miniconda, das uns bei der Verwaltung unserer Python-Umgebung hilft:
curl -O https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-arm64.sh
sh Miniconda3-latest-MacOSX-arm64.sh
Erstellen Sie nun eine dedizierte Umgebung für Flux:
conda create -n flux python=3.11 -y
conda activate flux
Installieren Sie PyTorch mit MPS (Metal Performance Shaders) Unterstützung:
pip install torch==2.3.1
Installieren Sie die erforderlichen Bibliotheken:
pip install git+https://github.com/huggingface/diffusers.git
pip install transformers==4.43.3 sentencepiece==0.2.0 accelerate==0.33.0 protobuf==5.27.3
Jetzt, da alles installiert ist, gehen wir den Prozess durch, um Flux Schnell auszuführen und ein Bild zu generieren.
Erstellen Sie eine neue Python-Datei namens flux_generate.py und fügen Sie den folgenden Code hinzu:
import torch
from diffusers import FluxPipeline
import diffusers
# Die rope-Funktion anpassen, um MPS-Gerät zu verarbeiten
_flux_rope = diffusers.models.transformers.transformer_flux.rope
def new_flux_rope(pos: torch.Tensor, dim: int, theta: int) -> torch.Tensor:
assert dim % 2 == 0, "Die Dimension muss gerade sein."
if pos.device.type == "mps":
return _flux_rope(pos.to("cpu"), dim, theta).to(device=pos.device)
else:
return _flux_rope(pos, dim, theta)
diffusers.models.transformers.transformer_flux.rope = new_flux_rope
# Das Flux Schnell-Modell laden
pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", revision='refs/pr/1', torch_dtype=torch.bfloat16).to("mps")
# Die Vorgabe für die Bildgenerierung festlegen
prompt = "Eine Katze, die ein Schild mit 'Hallo Welt' hält."
# Das Bild generieren
out = pipe(
prompt=prompt,
guidance_scale=0.,
height=1024,
width=1024,
num_inference_steps=4,
max_sequence_length=256,
).images[0]
# Das generierte Bild speichern
out.save("flux_image.png")
Wechseln Sie im Terminal zum Verzeichnis, das flux_generate.py enthält, und führen Sie aus:
python flux_generate.py
Dieser Vorgang sollte etwa 30 Sekunden dauern und bis zu 40 GB RAM auf Ihrem MacBook Pro M3 Max nutzen.
Zerlegen wir die wichtigsten Komponenten des Skripts:
Bibliotheken importieren: Wir importieren die erforderlichen Module von PyTorch und Diffusers.
Anpassen der ROPE-Funktion: Die Funktion new_flux_rope ist ein Workaround, um das MPS-Gerät (Metal Performance Shaders) zu behandeln. Damit wird die Kompatibilität mit dem M3 Max Chip gewährleistet.
Das Modell laden: Wir verwenden FluxPipeline.from_pretrained(), um das Flux Schnell-Modell zu laden, und geben das Modell-Repository und die Revision an.
Die Vorgabe festlegen: Wir definieren die Textvorgabe, die das zu generierende Bild beschreibt.
Das Bild generieren: Die Funktion pipe() wird mit verschiedenen Parametern aufgerufen:
prompt: Die Textbeschreibung des Bildesguidance_scale: Für dieses Beispiel auf 0 gesetztheight und width: Bildabmessungen (1024x1024 Pixel)num_inference_steps: Anzahl der Denoising-Schritte (in diesem Fall 4)max_sequence_length: Maximale Länge der Eingabesequenz (256)Das Bild speichern: Das generierte Bild wird als "flux_image.png" im selben Verzeichnis gespeichert.
Um die beste Leistung von Flux Schnell auf Ihrem MacBook Pro M3 Max zu erzielen:
num_inference_steps: Eine Erhöhung dieses Werts kann die Bildqualität verbessern, verlängert jedoch auch die Generierungszeit.max_sequence_length an: Längere Sequenzen ermöglichen detaillierte Vorgaben, erfordern jedoch mehr Speicher und Verarbeitungszeit.Wenn Sie auf Probleme stoßen:
to("mps")) anstelle von CUDA verwenden, da der M3 Max CUDA nicht unterstützt.max_sequence_length zu reduzieren, wenn Ihnen der Speicher ausgeht.Jetzt, da Flux Schnell lokal läuft, hier sind einige Ideen zum Erkunden:
guidance_scale, num_inference_steps und Bildabmessungen, um deren Auswirkungen auf die Ausgabe zu sehen.Die lokale Ausführung von Flux Schnell auf Ihrem MacBook Pro M3 Max eröffnet Ihnen eine Welt kreativer Möglichkeiten. Mit seiner beeindruckenden Geschwindigkeit und Qualität können Sie beeindruckende Bilder direkt auf Ihrem eigenen Gerät generieren. Wenn Sie sich mit dem Modell und seinen Parametern vertrauter machen, können Sie Ihre Ergebnisse feinabstimmen und die Grenzen der KI-generierten Kunst überschreiten.
Beachten Sie beim Verwenden von KI-generierten Bildern Urheberrechts- und ethische Aspekte und geben Sie immer dem Modell und seinen Erstellern Anerkennung, wenn Sie Ihre Ergebnisse teilen.
