proper usage picture and fix model loading

2023-10-23 14:21:53 +03:00
parent fc96bd51c0
commit b22586a423
2 changed files with 3 additions and 3 deletions
@@ -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