main

分支 (26)

标签 (1)

管理

管理

main

fix-progressbar-and-broken-theeye-link

add-portraitgenerator001

palmweaver-laion2b-kaliyuga-models

temporary-warp-notebook-fixes

temporary-warp-with-consistencychecker-code-under-gpl

symmetry-and-vit14_336px

fix_init_scale_bug

comment-out-downgrading-t4-v100-code

cleanup-of-torch-downgrade-and-indentation

changelog-updates-2022-04-10

fix-vr-translation-scale-issue

bugfixes-for-initial-vr-pr

allow-lowres-3d-animations

fix-disco-xform-utils-import-error

fix-video-input-ffmpeg-call-and-improve-imports

fix-adabins-model-installation

apply-import-change-to-ipynb

applying-removal-of-timestep-config-to-ipynb

fix-clone-of-pytorch3d-lite-in-ipynb

v5.4.0

disco-diffusion
/
disco_utils.py

import subprocess
from importlib import util as importlibutil

def module_exists(module_name):
  return importlibutil.find_spec(module_name)

def gitclone(url, targetdir=None):
    if targetdir:
        res = subprocess.run(['git', 'clone', url, targetdir], stdout=subprocess.PIPE).stdout.decode('utf-8')
    else:
        res = subprocess.run(['git', 'clone', url], stdout=subprocess.PIPE).stdout.decode('utf-8')
    print(res)

def pipi(modulestr):
  res = subprocess.run(['pip', 'install', modulestr], stdout=subprocess.PIPE).stdout.decode('utf-8')
  print(res)

def pipie(modulestr):
  res = subprocess.run(['git', 'install', '-e', modulestr], stdout=subprocess.PIPE).stdout.decode('utf-8')
  print(res)

def wget(url, outputdir):
  res = subprocess.run(['wget', url, '-P', f'{outputdir}'], stdout=subprocess.PIPE).stdout.decode('utf-8')
  print(res)