Update README.md

README.md

@@ -52,10 +52,8 @@ from schedulers.lcm_single_step_scheduler import LCMSingleStepScheduler
 from module.ip_adapter.utils import load_adapter_to_pipe
 from pipelines.sdxl_instantir import InstantIRPipeline
 
-# prepare models under ./checkpoints
-dcp_adapter = f'./models/adapter.pt'
-previewer_lora_path = f'./models'
-instantir_path = f'./models/aggregator.pt'
+# prepare models under ./models
+instantir_path = f'./models'
 
 # load pretrained models
 pipe = InstantIRPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', torch_dtype=torch.float16)
@@ -63,17 +61,17 @@ pipe = InstantIRPipeline.from_pretrained('stabilityai/stable-diffusion-xl-base-1
 # load adapter
 load_adapter_to_pipe(
     pipe,
-    dcp_adapter,
+    f"{instantir_path}/adapter.pt",
     image_encoder_or_path = 'facebook/dinov2-large',
 )
 
 # load previewer lora
-pipe.prepare_previewers(previewer_lora_path)
+pipe.prepare_previewers(instantir_path)
 pipe.scheduler = DDPMScheduler.from_pretrained('stabilityai/stable-diffusion-xl-base-1.0', subfolder="scheduler")
 lcm_scheduler = LCMSingleStepScheduler.from_config(pipe.scheduler.config)
 
 # load aggregator weights
-pretrained_state_dict = torch.load(instantir_path)
+pretrained_state_dict = torch.load(f"{instantir_path}/aggregator.pt")
 pipe.aggregator.load_state_dict(pretrained_state_dict)
 
 # send to GPU and fp16