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

fix library problems in preload_modules

Browse Source
This commit is contained in:
Lincoln Stein 2022-11-01 14:34:23 -04:00
parent ab2b5a691d
commit ef24d76adc
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
scripts/preload_models.py
View File

@ -6,6 +6,7 @@
#
# Coauthor: Kevin Turner http://github.com/keturn
#
print('Loading Python libraries...\n')
import argparse
import sys
import os
@ -17,15 +18,14 @@ from pathlib import Path
import traceback
import getpass
import requests
import clip
import transformers
import torch
transformers.logging.set_verbosity_error()
# deferred loading so that help message can be printed quickly
def load_libs():
print('Loading Python libraries...\n')
import clip
import transformers
import torch
import zipfile
transformers.logging.set_verbosity_error()
pass
#--------------------------globals--
Model_dir = './models/ldm/stable-diffusion-v1/'
@ -473,12 +473,14 @@ def download_codeformer():
#---------------------------------------------
def download_clipseg():
print('Installing clipseg model for text-based masking...',end='')
import zipfile
try:
model_url = 'https://owncloud.gwdg.de/index.php/s/ioHbRzFx6th32hn/download'
model_dest = 'src/clipseg/clipseg_weights.zip'
weights_dir = 'src/clipseg/weights'
if not os.path.exists(weights_dir):
os.makedirs(os.path.dirname(model_dest), exist_ok=True)
if not os.path.exists('src/clipseg/weights/rd64-uni-refined.pth'):
request.urlretrieve(model_url,model_dest)
with zipfile.ZipFile(model_dest,'r') as zip:
zip.extractall('src/clipseg')