Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Contribute to GitLab
Sign in / Register
Toggle navigation
S
stable-diffusion-webui
Project
Project
Details
Activity
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Board
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Administrator
stable-diffusion-webui
Commits
a2ae5a65
Commit
a2ae5a65
authored
Nov 24, 2022
by
Tiago F. Santos
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
[interrogator] mkdir check
parent
745f1e8f
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
10 additions
and
6 deletions
+10
-6
interrogate.py
modules/interrogate.py
+10
-6
No files found.
modules/interrogate.py
View file @
a2ae5a65
...
...
@@ -14,7 +14,8 @@ import modules.shared as shared
from
modules
import
devices
,
paths
,
lowvram
blip_image_eval_size
=
384
blip_model_local
=
os
.
path
.
join
(
'models'
,
'Interrogator'
,
'BLIP_model.pth'
)
blip_local_dir
=
os
.
path
.
join
(
'models'
,
'Interrogator'
)
blip_local_file
=
os
.
path
.
join
(
blip_local_dir
,
'model_base_caption_capfilt_large.pth'
)
blip_model_url
=
'https://storage.googleapis.com/sfr-vision-language-research/BLIP/models/model_base_caption_capfilt_large.pth'
clip_model_name
=
'ViT-L/14'
...
...
@@ -48,13 +49,16 @@ class InterrogateModels:
def
load_blip_model
(
self
):
import
models.blip
if
not
os
.
path
.
isfile
(
blip_model_local
):
if
not
os
.
path
.
isfile
(
blip_local_file
):
if
not
os
.
path
.
isdir
(
blip_local_dir
):
os
.
mkdir
(
blip_local_dir
)
print
(
"Downloading BLIP..."
)
import
requests
as
req
open
(
blip_
model_local
,
'wb'
)
.
write
(
req
.
get
(
blip_model_url
,
allow_redirects
=
True
)
.
content
)
print
(
"BLIP downloaded to"
,
blip_
model_local
+
'.'
)
from
requests
import
get
as
reqget
open
(
blip_
local_file
,
'wb'
)
.
write
(
req
get
(
blip_model_url
,
allow_redirects
=
True
)
.
content
)
print
(
"BLIP downloaded to"
,
blip_
local_file
+
'.'
)
blip_model
=
models
.
blip
.
blip_decoder
(
pretrained
=
blip_
model_local
,
image_size
=
blip_image_eval_size
,
vit
=
'base'
,
med_config
=
os
.
path
.
join
(
paths
.
paths
[
"BLIP"
],
"configs"
,
"med_config.json"
))
blip_model
=
models
.
blip
.
blip_decoder
(
pretrained
=
blip_
local_file
,
image_size
=
blip_image_eval_size
,
vit
=
'base'
,
med_config
=
os
.
path
.
join
(
paths
.
paths
[
"BLIP"
],
"configs"
,
"med_config.json"
))
blip_model
.
eval
()
return
blip_model
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment