Upload folder using huggingface_hub

.ipynb_checkpoints/README-checkpoint.md

@@ -0,0 +1,130 @@
+# Step1X-Anime-Edit-Lora
+
+This repository provides LoRA (Low-Rank Adaptation) support for the Step1X-Edit anime image editing model. It allows for fine-tuned control over image editing outputs.
+
+## Installation
+
+Refer to the main Step1X-Edit installation instructions at:  
+https://github.com/stepfun-ai/Step1X-Edit
+
+```bash
+wget https://huggingface.co/stepfun-ai/Step1X-Edit/resolve/main/step1x-edit-i1258.safetensors
+wget https://huggingface.co/stepfun-ai/Step1X-Edit/resolve/main/vae.safetensors
+huggingface-cli download Qwen/Qwen2.5-VL-7B-Instruct --local-dir Qwen2.5-VL-7B-Instruct
+```
+
+## Usage Examples
+
+### Basic Setup
+```python
+from inference import *
+
+image_edit = ImageGenerator(
+    ae_path="vae.safetensors",
+    dit_path="step1x-edit-i1258.safetensors",
+    qwen2vl_model_path='Qwen2.5-VL-7B-Instruct',
+    max_length=640,
+    quantized=True,
+    offload=True,
+    lora="change_output/step1x-edit_change-step00003000.safetensors",
+    mode="flash"
+)
+```
+
+### Example 1: Changing Background and Adding Elements
+```python
+image_path = "万叶.png"
+prompt = '''
+将背景改成公园，添加一些小松鼠
+'''
+
+num_steps = 28
+cfg_guidance = 4.5
+seed = 42
+size_level = 512  # Can also be 768 or 1024
+
+image = image_edit.generate_image(
+    prompt,
+    negative_prompt="",
+    ref_images=Image.open(image_path).convert("RGB"),
+    num_samples=1,
+    num_steps=num_steps,
+    cfg_guidance=cfg_guidance,
+    seed=seed,
+    show_progress=True,
+    size_level=size_level,
+)[0]
+image.save("万叶在公园.png")
+```
+
+-original output
+
+-lora output
+
+
+### Example 2: Advanced Scene Modification
+```python
+image_path = "万叶.png"
+prompt = '''
+将背景改成公园，添加一些小松鼠，天气为黄昏，调整为橙色光照，让男孩微笑
+'''
+
+# Same parameters as above
+image = image_edit.generate_image(...)
+image.save("万叶在黄昏.png")
+```
+
+-original output
+
+-lora output
+
+
+### Example 3: Character Modification
+```python
+image_path = "丝柯克.jpg"
+prompt = '''
+将图片背景变成海边，手里拿着一个冰淇凌
+'''
+
+num_steps = 28
+cfg_guidance = 6  # Higher guidance for more complex changes
+seed = 42
+size_level = 512
+
+image = image_edit.generate_image(...)
+image.save("丝柯克在海边.jpg")
+```
+
+-original output
+
+-lora output
+
+### Example 4: Object Replacement and Style Change
+```python
+image_path = "星铁海报.jpg"
+prompt = '''
+将桌子上的鞋替换成一个汉堡，背景换成星光咖啡厅，帽子换成小熊帽
+'''
+
+num_steps = 28
+cfg_guidance = 4.5
+seed = 42
+size_level = 512
+
+image = image_edit.generate_image(...)
+image.save("星铁小猫在咖啡厅.png")
+```
+
+-original output
+
+-lora output
+
+
+## Parameters
+- `num_steps`: Number of diffusion steps (typically 28)
+- `cfg_guidance`: Guidance scale (4.5-6 recommended)
+- `seed`: Random seed for reproducibility
+- `size_level`: Output resolution (512)
+
+## Output Comparison
+Each example shows the original output vs. LoRA-enhanced output for comparison.

README.md

@@ -0,0 +1,130 @@
+# Step1X-Anime-Edit-Lora
+
+This repository provides LoRA (Low-Rank Adaptation) support for the Step1X-Edit anime image editing model. It allows for fine-tuned control over image editing outputs.
+
+## Installation
+
+Refer to the main Step1X-Edit installation instructions at:  
+https://github.com/stepfun-ai/Step1X-Edit
+
+```bash
+wget https://huggingface.co/stepfun-ai/Step1X-Edit/resolve/main/step1x-edit-i1258.safetensors
+wget https://huggingface.co/stepfun-ai/Step1X-Edit/resolve/main/vae.safetensors
+huggingface-cli download Qwen/Qwen2.5-VL-7B-Instruct --local-dir Qwen2.5-VL-7B-Instruct
+```
+
+## Usage Examples
+
+### Basic Setup
+```python
+from inference import *
+
+image_edit = ImageGenerator(
+    ae_path="vae.safetensors",
+    dit_path="step1x-edit-i1258.safetensors",
+    qwen2vl_model_path='Qwen2.5-VL-7B-Instruct',
+    max_length=640,
+    quantized=True,
+    offload=True,
+    lora="change_output/step1x-edit_change-step00003000.safetensors",
+    mode="flash"
+)
+```
+
+### Example 1: Changing Background and Adding Elements
+```python
+image_path = "万叶.png"
+prompt = '''
+将背景改成公园，添加一些小松鼠
+'''
+
+num_steps = 28
+cfg_guidance = 4.5
+seed = 42
+size_level = 512  # Can also be 768 or 1024
+
+image = image_edit.generate_image(
+    prompt,
+    negative_prompt="",
+    ref_images=Image.open(image_path).convert("RGB"),
+    num_samples=1,
+    num_steps=num_steps,
+    cfg_guidance=cfg_guidance,
+    seed=seed,
+    show_progress=True,
+    size_level=size_level,
+)[0]
+image.save("万叶在公园.png")
+```
+
+-original output
+
+-lora output
+
+
+### Example 2: Advanced Scene Modification
+```python
+image_path = "万叶.png"
+prompt = '''
+将背景改成公园，添加一些小松鼠，天气为黄昏，调整为橙色光照，让男孩微笑
+'''
+
+# Same parameters as above
+image = image_edit.generate_image(...)
+image.save("万叶在黄昏.png")
+```
+
+-original output
+
+-lora output
+
+
+### Example 3: Character Modification
+```python
+image_path = "丝柯克.jpg"
+prompt = '''
+将图片背景变成海边，手里拿着一个冰淇凌
+'''
+
+num_steps = 28
+cfg_guidance = 6  # Higher guidance for more complex changes
+seed = 42
+size_level = 512
+
+image = image_edit.generate_image(...)
+image.save("丝柯克在海边.jpg")
+```
+
+-original output
+
+-lora output
+
+### Example 4: Object Replacement and Style Change
+```python
+image_path = "星铁海报.jpg"
+prompt = '''
+将桌子上的鞋替换成一个汉堡，背景换成星光咖啡厅，帽子换成小熊帽
+'''
+
+num_steps = 28
+cfg_guidance = 4.5
+seed = 42
+size_level = 512
+
+image = image_edit.generate_image(...)
+image.save("星铁小猫在咖啡厅.png")
+```
+
+-original output
+
+-lora output
+
+
+## Parameters
+- `num_steps`: Number of diffusion steps (typically 28)
+- `cfg_guidance`: Guidance scale (4.5-6 recommended)
+- `seed`: Random seed for reproducibility
+- `size_level`: Output resolution (512)
+
+## Output Comparison
+Each example shows the original output vs. LoRA-enhanced output for comparison.

change_output/step1x-edit_change-step00000500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:51e215ee520a4d4085f752725c1d74ff815ce88013e531faa62b39b15aff9072
+size 612746080

change_output/step1x-edit_change-step00001000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec22b4aba8c0b5fb77b2a631951ceaf44b3cf00e26707d70166bab19d2eb2e4f
+size 612746080

change_output/step1x-edit_change-step00001500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2a7dafe623bdc32a8f2eb0af3d28d98a5a2d8756f64a32d8e1916c1748bf0db7
+size 612746080

change_output/step1x-edit_change-step00002000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:682852f81524a0d760ce1e4235341fc5e16dcdb563fa5fb189aa3ff40122f141
+size 612746080

change_output/step1x-edit_change-step00002500.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9b70032ae5ce434dd17ab8e20046ae57f743d9b102d0dc2333a69f722ed7dce9
+size 612746080

change_output/step1x-edit_change-step00003000.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:230002d05703490668c99005ceab891faebea74b8d89e7af9695b57719194c1b
+size 612746080