1580158
/
sd-webui-text2video
mirror of https://github.com/kabachuha/sd-webui-text2video.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

model search path fix

pull/186/head
kabachuha 2023-07-02 18:44:43 +03:00
parent c82362f227
commit 7f2a26a114
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
scripts/t2v_helpers/general_utils.py
View File

@ -18,11 +18,11 @@ def get_model_location(model_name):
assert model_name is not None
if model_name == "<modelscope>":
return os.path.join(ph.models_path, 'models/ModelScope/t2v')
return os.path.join(ph.models_path, 'ModelScope/t2v')
elif model_name == "<videocrafter>":
return os.path.join(ph.models_path, 'models/VideoCrafter')
return os.path.join(ph.models_path, 'VideoCrafter')
else:
return os.path.join(ph.models_path, 'models/text2video/', model_name)
return os.path.join(ph.models_path, 'text2video/', model_name)
def reconstruct_conds(cond, uncond, step):
c = reconstruct_cond_batch(cond, step)