lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-04-30 03:01:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

revise space

Browse Source
This commit is contained in:
layerdiffusion
2024-08-19 22:43:09 -07:00
parent fef6df29d9
commit 65ec461f8a
9 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
extensions-builtin/forge_space_birefnet/forge_app.py
View File

@@ -11,7 +11,7 @@ from torchvision import transforms
os.environ['HOME'] = spaces.convert_root_path() + 'home'
with spaces.GPUObject() as birefnet_gpu_obj:
with spaces.capture_gpu_object() as birefnet_gpu_obj:
birefnet = AutoModelForImageSegmentation.from_pretrained(
"ZhengPeng7/BiRefNet", trust_remote_code=True
)