Deeper disable of --nobreakmodel attempt
This commit is contained in:
parent
45495d8792
commit
bddcd7ab7f
|
@ -545,6 +545,10 @@ def device_config(config):
|
||||||
n_layers = utils.num_layers(config)
|
n_layers = utils.num_layers(config)
|
||||||
if args.cpu:
|
if args.cpu:
|
||||||
breakmodel.gpu_blocks = [0]*n_layers
|
breakmodel.gpu_blocks = [0]*n_layers
|
||||||
|
return
|
||||||
|
elif vars.nobreakmodel:
|
||||||
|
breakmodel.gpu_blocks = [0]*n_layers
|
||||||
|
return
|
||||||
elif(args.breakmodel_gpulayers is not None or (utils.HAS_ACCELERATE and args.breakmodel_disklayers is not None)):
|
elif(args.breakmodel_gpulayers is not None or (utils.HAS_ACCELERATE and args.breakmodel_disklayers is not None)):
|
||||||
try:
|
try:
|
||||||
if(not args.breakmodel_gpulayers):
|
if(not args.breakmodel_gpulayers):
|
||||||
|
|
Loading…
Reference in New Issue