diff --git a/README.MD b/README.MD
index 42760be..0d78f7f 100644
--- a/README.MD
+++ b/README.MD
@@ -685,7 +685,7 @@ If the folder set in ini is invalid, the font folder that comes with the plugin
### LaMa

Erase objects from the image based on the mask. this node is repackage of [IOPaint](https://www.iopaint.com), powered by state-of-the-art AI models.
-It is have [LaMa](https://github.com/advimman/lama), [LDM](https://github.com/CompVis/latent-diffusion), [ZITS](https://github.com/DQiaole/ZITS_inpainting),[MAT](https://github.com/fenglinglwb/MAT), [FcF](https://github.com/SHI-Labs/FcF-Inpainting), [Manga](https://github.com/msxie92/MangaInpainting) models and the OpenCV method to erase. Please refer to the original link for the introduction of each model.
+It is have [LaMa](https://github.com/advimman/lama), [LDM](https://github.com/CompVis/latent-diffusion), [ZITS](https://github.com/DQiaole/ZITS_inpainting),[MAT](https://github.com/fenglinglwb/MAT), [FcF](https://github.com/SHI-Labs/FcF-Inpainting), [Manga](https://github.com/msxie92/MangaInpainting) models and the SPREAD method to erase. Please refer to the original link for the introduction of each model.
Please download the model files from [lama models(Baidu Netdisk)](https://pan.baidu.com/s/1LllR9TJHP1G9uEwWT3Mvkg?pwd=tvzv) or [lama models(Google Drive)](https://drive.google.com/drive/folders/1Aq0a4sybb3SRxi7j1e1_ZbBRjaWDdP9e?usp=sharing) to ```ComfyUI/models/lama``` folder.
Node optons:
diff --git a/README_CN.MD b/README_CN.MD
index 1f68b63..2ce74e0 100644
--- a/README_CN.MD
+++ b/README_CN.MD
@@ -677,7 +677,7 @@ GradientImage的V2升级版。
### LaMa

根据图像遮罩擦除物体。本节点是对[IOPaint](https://www.iopaint.com)的封装,由 SOTA AI 模型提供支持。
-提供[LaMa](https://github.com/advimman/lama), [LDM](https://github.com/CompVis/latent-diffusion), [ZITS](https://github.com/DQiaole/ZITS_inpainting),[MAT](https://github.com/fenglinglwb/MAT), [FcF](https://github.com/SHI-Labs/FcF-Inpainting), [Manga](https://github.com/msxie92/MangaInpainting) 模型以及 OpenCV 擦除方法。请查看链接了解各个模型的介绍。
+提供[LaMa](https://github.com/advimman/lama), [LDM](https://github.com/CompVis/latent-diffusion), [ZITS](https://github.com/DQiaole/ZITS_inpainting),[MAT](https://github.com/fenglinglwb/MAT), [FcF](https://github.com/SHI-Labs/FcF-Inpainting), [Manga](https://github.com/msxie92/MangaInpainting) 模型以及 SPREAD 擦除方法。请查看链接了解各个模型的介绍。
请下载模型文件 [lama models(百度网盘)](https://pan.baidu.com/s/1LllR9TJHP1G9uEwWT3Mvkg?pwd=tvzv) 或者 [lama models(Google Drive)](https://drive.google.com/drive/folders/1Aq0a4sybb3SRxi7j1e1_ZbBRjaWDdP9e?usp=sharing), 将文件放到```ComfyUI/models/lama```
节点选项说明:
diff --git a/api_key.ini b/api_key.ini
index aefa497..71424fa 100644
--- a/api_key.ini
+++ b/api_key.ini
@@ -1,2 +1,2 @@
# LayerStyle api_key
-google_api_key=
\ No newline at end of file
+google_api_key=AIzaSyCJA0LNM7GwVmO6UZG29sjQLe2D40klyr8
\ No newline at end of file
diff --git a/image/lama_example.png b/image/lama_example.png
index 3cc11c3..bce050a 100644
Binary files a/image/lama_example.png and b/image/lama_example.png differ
diff --git a/py/lama.py b/py/lama.py
index ed95e45..eb2c05f 100644
--- a/py/lama.py
+++ b/py/lama.py
@@ -14,7 +14,7 @@ class LaMa:
@classmethod
def INPUT_TYPES(self):
- model_list = ['lama', 'ldm', 'zits', 'mat', 'fcf', 'manga', 'opencv']
+ model_list = ['lama', 'ldm', 'zits', 'mat', 'fcf', 'manga', 'spread']
device_list = ['cuda', 'cpu']
return {
"required": {
@@ -67,7 +67,7 @@ class LaMa:
if mask_grow or mask_blur:
_mask = tensor2pil(expand_mask(image2mask(_mask), mask_grow, mask_blur))
- if lama_model == 'opencv':
+ if lama_model == 'spread':
ret_image = pixel_spread(tensor2pil(_image).convert('RGB'), ImageChops.invert(_mask.convert('RGB')))
else:
temp_dir = os.path.join(folder_paths.get_temp_directory(), generate_random_name('_lama_', '_temp', 16))
diff --git a/workflow/lama_example.json b/workflow/lama_example.json
index c480b00..edfd303 100644
--- a/workflow/lama_example.json
+++ b/workflow/lama_example.json
@@ -603,7 +603,7 @@
"Node name for S&R": "LayerUtility: LaMa"
},
"widgets_values": [
- "opencv",
+ "spread",
"cuda",
false,
25,