Update README.md
Browse files
README.md
CHANGED
@@ -72,7 +72,7 @@ from transformers import Qwen2_5_VLForConditionalGeneration, Qwen2_5_VLProcessor
|
|
72 |
from diffusers import FlowMatchEulerDiscreteScheduler, AutoencoderKL
|
73 |
|
74 |
# Load model components
|
75 |
-
pretrained_model_name_or_path = "
|
76 |
|
77 |
transformer = SD3Transformer2DKontextModel.from_pretrained(
|
78 |
pretrained_model_name_or_path, subfolder="transformer", torch_dtype=torch.bfloat16).cuda()
|
@@ -106,7 +106,7 @@ pipeline = StableDiffusion3KontextPipeline(
|
|
106 |
|
107 |
# Prepare prompts
|
108 |
prompt = 'a pig with wings and a top hat flying over a happy futuristic scifi city'
|
109 |
-
negative_prompt = ''
|
110 |
|
111 |
messages = [[{"role": "user", "content": [{"type": "text", "text": f'Generate an image: {txt}'}]}]
|
112 |
for txt in [prompt, negative_prompt]]
|
|
|
72 |
from diffusers import FlowMatchEulerDiscreteScheduler, AutoencoderKL
|
73 |
|
74 |
# Load model components
|
75 |
+
pretrained_model_name_or_path = "Skywork/UniPic2-Metaquery-GRPO-9B"
|
76 |
|
77 |
transformer = SD3Transformer2DKontextModel.from_pretrained(
|
78 |
pretrained_model_name_or_path, subfolder="transformer", torch_dtype=torch.bfloat16).cuda()
|
|
|
106 |
|
107 |
# Prepare prompts
|
108 |
prompt = 'a pig with wings and a top hat flying over a happy futuristic scifi city'
|
109 |
+
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.'
|
110 |
|
111 |
messages = [[{"role": "user", "content": [{"type": "text", "text": f'Generate an image: {txt}'}]}]
|
112 |
for txt in [prompt, negative_prompt]]
|