Update README.md
Browse files
README.md
CHANGED
|
@@ -10,7 +10,6 @@ import torch
|
|
| 10 |
from diffusers.utils import load_image
|
| 11 |
|
| 12 |
repo_id = "YiYiXu/modular-diffdiff"
|
| 13 |
-
|
| 14 |
components = ComponentsManager()
|
| 15 |
|
| 16 |
diffdiff_pipeline = ModularPipeline.from_pretrained(repo_id, trust_remote_code=True, component_manager=components, collection="diffdiff")
|
|
@@ -18,8 +17,6 @@ diffdiff_pipeline.load_componenets(torch_dtype=torch.float16)
|
|
| 18 |
|
| 19 |
components.enable_auto_cpu_offload()
|
| 20 |
|
| 21 |
-
|
| 22 |
-
|
| 23 |
image = load_image( "https://huggingface.co/datasets/OzzyGT/testing-resources/resolve/main/differential/20240329211129_4024911930.png?download=true")
|
| 24 |
mask = load_image("https://huggingface.co/datasets/OzzyGT/testing-resources/resolve/main/differential/gradient_mask.png?download=true")
|
| 25 |
|
|
|
|
| 10 |
from diffusers.utils import load_image
|
| 11 |
|
| 12 |
repo_id = "YiYiXu/modular-diffdiff"
|
|
|
|
| 13 |
components = ComponentsManager()
|
| 14 |
|
| 15 |
diffdiff_pipeline = ModularPipeline.from_pretrained(repo_id, trust_remote_code=True, component_manager=components, collection="diffdiff")
|
|
|
|
| 17 |
|
| 18 |
components.enable_auto_cpu_offload()
|
| 19 |
|
|
|
|
|
|
|
| 20 |
image = load_image( "https://huggingface.co/datasets/OzzyGT/testing-resources/resolve/main/differential/20240329211129_4024911930.png?download=true")
|
| 21 |
mask = load_image("https://huggingface.co/datasets/OzzyGT/testing-resources/resolve/main/differential/gradient_mask.png?download=true")
|
| 22 |
|