Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge branch 'development' into asymmetric-tiling

Browse Source
This commit is contained in:
Carson Katri 2022-10-18 22:37:18 -04:00 committed by GitHub
commit 15aa7593f6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ldm/invoke/txt2mask.py
View File

@ -29,7 +29,7 @@ work fine.
import torch import torch
import numpy as np import numpy as np
from models.clipseg import CLIPDensePredT from clipseg import CLIPDensePredT
from einops import rearrange, repeat from einops import rearrange, repeat
from PIL import Image from PIL import Image
from torchvision import transforms from torchvision import transforms

3
scripts/preload_models.py
View File

@ -117,7 +117,7 @@ try:
with zipfile.ZipFile(model_dest,'r') as zip: with zipfile.ZipFile(model_dest,'r') as zip:
zip.extractall('src/clipseg') zip.extractall('src/clipseg')
os.rename('src/clipseg/clipseg_weights','src/clipseg/weights') os.rename('src/clipseg/clipseg_weights','src/clipseg/weights')
from models.clipseg import CLIPDensePredT from clipseg import CLIPDensePredT
model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, ) model = CLIPDensePredT(version='ViT-B/16', reduce_dim=64, )
model.eval() model.eval()
model.load_state_dict( model.load_state_dict(
@ -126,6 +126,7 @@ try:
map_location=torch.device('cpu'), map_location=torch.device('cpu'),
strict=False, strict=False,
) )
)
except Exception: except Exception:
print('Error installing clipseg model:') print('Error installing clipseg model:')
print(traceback.format_exc()) print(traceback.format_exc())