Update Lama.py

This commit is contained in:
shadowcz007
2023-12-31 21:24:40 +08:00
parent 9ae8d30b6b
commit 024ff4a309
+1 -1
View File
@@ -12,7 +12,7 @@ import torch
llma_model_path=os.path.join(folder_paths.models_dir, "lama/big-lama.pt")
if not os.path.exists(llma_model_path):
os.environ['LAMA_MODEL']=''
print(f"lama torchscript model not found: {llma_model_path}")
print(f"## lama torchscript model not found: {llma_model_path},pls download from https://github.com/enesmsahin/simple-lama-inpainting/releases/download/v0.1.0/big-lama.pt")
else:
os.environ['LAMA_MODEL'] = llma_model_path