ImageReward fix, #214

pull/218/head
AlUlkesh 2023-07-16 09:15:29 +02:00
parent 9229ed5e7e
commit c6fcf6c438
1 changed files with 3 additions and 2 deletions

View File

@ -6,12 +6,13 @@ if not launch.is_installed("send2trash"):
try: try:
import ImageReward import ImageReward
import dataset import datasets
import dill import dill
import diffusers import diffusers
import multiprocessing import multiprocessing
import pyarrow import pyarrow
import xxhash import xxhash
except ImportError as e: except (ImportError, ModuleNotFoundError) as e:
#print(e)
req_IR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "req_IR.txt") req_IR = os.path.join(os.path.dirname(os.path.realpath(__file__)), "req_IR.txt")
launch.run_pip(f'install -r "{req_IR}" --no-deps image-reward', 'ImageReward requirement for image browser') launch.run_pip(f'install -r "{req_IR}" --no-deps image-reward', 'ImageReward requirement for image browser')