Upload README.md with huggingface_hub

README.md

@@ -46,22 +46,19 @@ import torch
 model_name = "OPPOer/Qwen-Image-Edit-Pruning/Qwen-Image-Edit-13B-4steps"
 pipe = QwenImageEditPipeline.from_pretrained(model_name, torch_dtype=torch.bfloat16)
 pipe = pipe.to('cuda')
-output_path = 'outputs'
-os.makedirs(output_path, exist_ok=True)
-for file_name in os.listdir('examples'):
-    prompt = file_name.replace('_in.jpg', '')
-    subject_img = Image.open(os.path.join('examples', file_name)).convert('RGB')
-    t1 = time.time()
-    inputs = {
-        "image": subject_img,
-        "prompt": prompt,
-        "generator": torch.manual_seed(42),
-        "true_cfg_scale": 1,
-        "num_inference_steps": 4,
-    }
-    with torch.inference_mode():
-        output = pipe(**inputs)
-        output_image = output.images[0]
-        output_image.save(os.path.join(output_path, f'{prompt}.jpg'))
-    print(time.time()-t1)
+
+subject_img = Image.open('input.jpg').convert('RGB')
+prompt = '数字插画风格'
+t1 = time.time()
+inputs = {
+    "image": subject_img,
+    "prompt": prompt,
+    "generator": torch.manual_seed(42),
+    "true_cfg_scale": 1,
+    "num_inference_steps": 4,
+}
+with torch.inference_mode():
+    output = pipe(**inputs)
+    output_image = output.images[0]
+    output_image.save('output.jpg')
 ```