mlii0117 commited on
Commit
61b016d
·
verified ·
1 Parent(s): b9fa26c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -38,7 +38,7 @@ from diffusers import StableDiffusionInstructPix2PixPipeline
38
  import torch
39
 
40
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
41
- "TUMSyn/ct-noncontrast-to-contrast", torch_dtype=torch.float16
42
  ).to("cuda")
43
  generator = torch.Generator("cuda").manual_seed(0)
44
  ```
 
38
  import torch
39
 
40
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
41
+ "mlii0117/sd1.5_MPECT", torch_dtype=torch.float16
42
  ).to("cuda")
43
  generator = torch.Generator("cuda").manual_seed(0)
44
  ```