Update README.md

README.md

@@ -31,15 +31,13 @@ import torch
 # Load the fine-tuned UNet
 unet = UNet2DConditionModel.from_pretrained(
     "ezlee258258/Inversion-DPO",
-    subfolder="unet",
-    torch_dtype=torch.float16
+    subfolder="unet"
 )
 
 # Load the pipeline with the fine-tuned UNet
 pipe = StableDiffusionXLPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",
-    unet=unet,
-    torch_dtype=torch.float16
+    unet=unet
 )
 pipe = pipe.to("cuda")