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

resolve models.clipseg vs clipseg ambiguity

Browse Source
This commit is contained in:
Lincoln Stein 2022-10-18 23:09:26 -04:00
parent 9b3ac92c24
commit 62dd3b7d7d
2 changed files with 2 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 clipseg import CLIPDensePredT from models.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

2
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 clipseg import CLIPDensePredT from models.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(