xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-09-09 10:37:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

remove logs

Browse Source
This commit is contained in:
drunkplato 2025-01-14 20:10:34 +00:00 committed by Ubuntu
parent 46dea2d5e4
commit 47ee984278
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
custom_nodes/MemedeckComfyNodes/nodes_preprocessing.py
View File

@ -113,10 +113,10 @@ class MD_ImageToMotionPrompt:
), ),
"max_tokens": ("INT", {"min": 1, "max": 2048, "default": 200}), "max_tokens": ("INT", {"min": 1, "max": 2048, "default": 200}),
}, },
# "optional": { "optional": {
# "temperature": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.2}), "temperature": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.2}),
# "top_p": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.9}), "top_p": ("FLOAT", {"min": 0.0, "max": 1.0, "step": 0.01, "default": 0.7}),
# } }
} }
@ -127,9 +127,9 @@ class MD_ImageToMotionPrompt:
def generate_completion( def generate_completion(
self, pre_prompt: str, post_prompt: str, Image: torch.Tensor, clip, prompt: str, negative_prompt: str, self, pre_prompt: str, post_prompt: str, Image: torch.Tensor, clip, prompt: str, negative_prompt: str,
# temperature: float, temperature: float = 0.2,
# top_p: float, top_p: float = 0.7,
max_tokens: int max_tokens: int = 256
) -> Tuple[str]: ) -> Tuple[str]:
# start a timer # start a timer
start_time = time.time() start_time = time.time()
@ -139,8 +139,8 @@ class MD_ImageToMotionPrompt:
response = requests.post("http://127.0.0.1:5010/inference", json={ response = requests.post("http://127.0.0.1:5010/inference", json={
"image_url": f"data:image/jpeg;base64,{b64image}", "image_url": f"data:image/jpeg;base64,{b64image}",
"prompt": prompt, "prompt": prompt,
"temperature": 0.2, "temperature": temperature,
"top_p": 0.7, "top_p": top_p,
"max_gen_len": max_tokens, "max_gen_len": max_tokens,
}) })
if response.status_code != 200: if response.status_code != 200:
@ -339,7 +339,6 @@ class MD_CompressAdjustNode:
image_cv2 = cv2.cvtColor(np.array(tensor2pil(image)), cv2.COLOR_RGB2BGR) image_cv2 = cv2.cvtColor(np.array(tensor2pil(image)), cv2.COLOR_RGB2BGR)
# calculate the crf based on the image # calculate the crf based on the image
analysis_results = self.analyze_compression_artifacts(image_cv2, width=width, height=height) analysis_results = self.analyze_compression_artifacts(image_cv2, width=width, height=height)
logger.info(f"compression analysis_results: {analysis_results}")
calculated_crf = self.calculate_crf(analysis_results, self.ideal_blockiness, self.ideal_edge_density, calculated_crf = self.calculate_crf(analysis_results, self.ideal_blockiness, self.ideal_edge_density,
self.ideal_color_variation, self.blockiness_weight, self.ideal_color_variation, self.blockiness_weight,
self.edge_density_weight, self.color_variation_weight) self.edge_density_weight, self.color_variation_weight)
@ -347,8 +346,6 @@ class MD_CompressAdjustNode:
if desired_crf is 0: if desired_crf is 0:
desired_crf = calculated_crf desired_crf = calculated_crf
logger.info(f"calculated_crf: {calculated_crf}")
logger.info(f"desired_crf: {desired_crf}")
args = [ args = [
utils.ffmpeg_path, utils.ffmpeg_path,
"-v", "error", "-v", "error",