OrlandoHugBot commited on
Commit
c7c806f
·
verified ·
1 Parent(s): 6e2c83e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "/path/to/unipicv2_qwen2_5_vl_7b_sd_3_5m_kontext"
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]]