mlii0117 commited on
Commit
5f73389
·
verified ·
1 Parent(s): 6baa8cc

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -70,7 +70,7 @@ from pydicom import dcmread
70
  from diffusers import StableDiffusionInstructPix2PixPipeline
71
 
72
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
73
- "TUMSyn/ct-noncontrast-to-contrast", torch_dtype=torch.float16
74
  ).to("cuda")
75
  generator = torch.Generator("cuda").manual_seed(0)
76
 
 
70
  from diffusers import StableDiffusionInstructPix2PixPipeline
71
 
72
  pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(
73
+ "mlii0117/sd1.5_MPECT", torch_dtype=torch.float16
74
  ).to("cuda")
75
  generator = torch.Generator("cuda").manual_seed(0)
76