BiliSakura commited on
Commit
de5f0e6
·
verified ·
1 Parent(s): 54aff06

Update all files for DiffusionSat-SR-Texas-256

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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="./custom_pipelines/pipeline_diffusionsat.py", # Path to this file
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="./custom_pipelines/pipeline_diffusionsat_controlnet.py", # Path to this file
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
  )