mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

silence printing the model because it's just useless noise

Browse Source
This commit is contained in:
mrq 2023-03-04 16:38:24 +00:00
parent 71cc43e65c
commit 6eb7ebf847
3 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/train.py
View File

@ -19,6 +19,7 @@ from datetime import datetime
from utils.util import opt_get, map_cuda_to_correct_device from utils.util import opt_get, map_cuda_to_correct_device
import tortoise.utils.torch_intermediary as ml
def init_dist(backend, **kwargs): def init_dist(backend, **kwargs):
# These packages have globals that screw with Windows, so only import them if needed. # These packages have globals that screw with Windows, so only import them if needed.

2
codes/trainer/ExtensibleTrainer.py
View File

@ -185,7 +185,7 @@ class ExtensibleTrainer(BaseModel):
self.env['discriminators'] = self.netsD self.env['discriminators'] = self.netsD
self.env['emas'] = self.emas self.env['emas'] = self.emas
self.print_network() # print network # self.print_network() # print network
self.load() # load networks from save states as needed self.load() # load networks from save states as needed
# Load experiments # Load experiments

3
requirements.txt
View File

@ -48,4 +48,5 @@ x-clip
x_transformers==1.0.4 x_transformers==1.0.4
# bitsandbytes # bitsandbytes
bitsandbytes==0.35.0 bitsandbytes==0.35.0 ; platform_system == "Windows"
bitsandbytes ; platform_system != "Windows"