Update all files for DiffusionSat-SR-Texas-256
Browse files
README.md
CHANGED
|
@@ -43,7 +43,7 @@ from diffusers import DiffusionPipeline
|
|
| 43 |
# Load pipeline
|
| 44 |
pipe = DiffusionPipeline.from_pretrained(
|
| 45 |
"path/to/ckpt/diffusionsat",
|
| 46 |
-
custom_pipeline="./
|
| 47 |
torch_dtype=torch.float16,
|
| 48 |
trust_remote_code=True,
|
| 49 |
)
|
|
@@ -79,7 +79,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
| 79 |
pipe = DiffusionPipeline.from_pretrained(
|
| 80 |
"path/to/ckpt/diffusionsat",
|
| 81 |
controlnet=controlnet,
|
| 82 |
-
custom_pipeline="./
|
| 83 |
torch_dtype=torch.float16,
|
| 84 |
trust_remote_code=True,
|
| 85 |
)
|
|
|
|
| 43 |
# Load pipeline
|
| 44 |
pipe = DiffusionPipeline.from_pretrained(
|
| 45 |
"path/to/ckpt/diffusionsat",
|
| 46 |
+
custom_pipeline="./pipeline_diffusionsat.py", # Path to this file
|
| 47 |
torch_dtype=torch.float16,
|
| 48 |
trust_remote_code=True,
|
| 49 |
)
|
|
|
|
| 79 |
pipe = DiffusionPipeline.from_pretrained(
|
| 80 |
"path/to/ckpt/diffusionsat",
|
| 81 |
controlnet=controlnet,
|
| 82 |
+
custom_pipeline="./pipeline_diffusionsat_controlnet.py", # Path to this file
|
| 83 |
torch_dtype=torch.float16,
|
| 84 |
trust_remote_code=True,
|
| 85 |
)
|