diff --git a/nodes.py b/nodes.py index f4e5a24..e8b5240 100644 --- a/nodes.py +++ b/nodes.py @@ -42,8 +42,8 @@ class LCMSampler: def sample(self, ckpt_name, seed, steps, cfg, positive_prompt, size, num_images): if self.pipe is None: self.pipe = LatentConsistencyModelPipeline.from_pretrained( - pretrained_model_name_or_path=folder_paths.get_annotated_filepath(ckpt_name, "checkpoints"), - local_files_only=True, + "SimianLuo/LCM_Dreamshaper_v7",#folder_paths.get_annotated_filepath(ckpt_name, "checkpoints"), + #local_files_only=True, scheduler=self.scheduler ) self.pipe.to(get_torch_device()) @@ -64,7 +64,7 @@ class LCMSampler: print("LCM inference time: ", time.time() - start_time, "seconds") - return (result,) + return ({"samples":result},) NODE_CLASS_MAPPINGS = { "LCMSampler": LCMSampler diff --git a/sample.png b/sample.png index c050e17..ef2c130 100644 Binary files a/sample.png and b/sample.png differ