nodes/Nojahhh-ComfyUI_GLM4_Wrapper
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Johan Mellin
2024-12-04 21:31:53 +01:00
committed by GitHub
parent b9052f6d82
commit e97cfa17c4
1 changed files with 6 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+6 -2
View File
@@ -93,6 +93,7 @@ Enhances a given prompt using the GLM-4 model.
- **repetition_penalty**: Repetition penalty for sampling.
- **image** (optional): Image to enhance the prompt. Only works with `THUDM/glm-4v-9b`, `alexwww94/glm-4v-9b-gptq-4bit` and `alexwww94/glm-4v-9b-gptq-3bit`.
- **unload_model**: Unload the model after use.
- **seed** (optional): Seed for reproducability.
#### Output
@@ -114,6 +115,7 @@ Performs inference using the GLM-4 model.
- **repetition_penalty**: Repetition penalty for sampling.
- **image** (optional): Image to use as input for inferencing. Only works with `THUDM/glm-4v-9b`, `alexwww94/glm-4v-9b-gptq-4bit` and `alexwww94/glm-4v-9b-gptq-3bit`.
- **unload_model**: Unload the model after use.
- **seed** (optional): Seed for reproducability.
#### Output
@@ -176,7 +178,8 @@ enhanced_prompt = enhancer.enhance_prompt(
top_p=0.7,
repetition_penalty=1.1,
image=None, # PIL Image
unload_model=True
unload_model=True,
seed=42
)
print(enhanced_prompt)
```
@@ -202,7 +205,8 @@ output_text = inference.infer(
top_p=1,
repetition_penalty=1.0,
image=None,
unload_model=True
unload_model=True,
seed=42
)
print(output_text)
```