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 numpy as np
from clipseg import CLIPDensePredT
from models.clipseg import CLIPDensePredT
from einops import rearrange, repeat
from PIL import Image
from torchvision import transforms

2
scripts/preload_models.py
View File

@ -117,7 +117,7 @@ try:
with zipfile.ZipFile(model_dest,'r') as zip:
zip.extractall('src/clipseg')
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.eval()
model.load_state_dict(