Update README.md
Browse files
README.md
CHANGED
@@ -80,7 +80,7 @@ from transformers import Qwen2_5_VLForConditionalGeneration, Qwen2_5_VLProcessor
|
|
80 |
from diffusers import FlowMatchEulerDiscreteScheduler, AutoencoderKL
|
81 |
|
82 |
# Load model components
|
83 |
-
pretrained_model_name_or_path = "/
|
84 |
|
85 |
transformer = SD3Transformer2DKontextModel.from_pretrained(
|
86 |
pretrained_model_name_or_path, subfolder="transformer", torch_dtype=torch.bfloat16).cuda()
|
@@ -114,7 +114,7 @@ pipeline = StableDiffusion3KontextPipeline(
|
|
114 |
|
115 |
# Prepare prompts
|
116 |
prompt = 'a pig with wings and a top hat flying over a happy futuristic scifi city'
|
117 |
-
negative_prompt = ''
|
118 |
|
119 |
messages = [[{"role": "user", "content": [{"type": "text", "text": f'Generate an image: {txt}'}]}]
|
120 |
for txt in [prompt, negative_prompt]]
|
|
|
80 |
from diffusers import FlowMatchEulerDiscreteScheduler, AutoencoderKL
|
81 |
|
82 |
# Load model components
|
83 |
+
pretrained_model_name_or_path = "Skywork/UniPic2-Metaquery-9B"
|
84 |
|
85 |
transformer = SD3Transformer2DKontextModel.from_pretrained(
|
86 |
pretrained_model_name_or_path, subfolder="transformer", torch_dtype=torch.bfloat16).cuda()
|
|
|
114 |
|
115 |
# Prepare prompts
|
116 |
prompt = 'a pig with wings and a top hat flying over a happy futuristic scifi city'
|
117 |
+
negative_prompt = 'blurry, low quality, low resolution, distorted, deformed, broken content, missing parts, damaged details, artifacts, glitch, noise, pixelated, grainy, compression artifacts, bad composition, wrong proportion, incomplete editing, unfinished, unedited areas.'
|
118 |
|
119 |
messages = [[{"role": "user", "content": [{"type": "text", "text": f'Generate an image: {txt}'}]}]
|
120 |
for txt in [prompt, negative_prompt]]
|