ironjr commited on
Commit
516cc24
1 Parent(s): 36e8307

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -98,8 +98,8 @@ device = f'cuda:{opt.device}' if opt.device >= 0 else 'cpu'
98
 
99
  if opt.model is None:
100
  # opt.model = 'cagliostrolab/animagine-xl-3.1'
101
- opt.model = 'ironjr/BlazingDriveV11m'
102
- # opt.model = 'KBlueLeaf/kohaku-v2.1'
103
  else:
104
  if opt.model.endswith('.safetensors'):
105
  opt.model = os.path.abspath(os.path.join('checkpoints', opt.model))
 
98
 
99
  if opt.model is None:
100
  # opt.model = 'cagliostrolab/animagine-xl-3.1'
101
+ # opt.model = 'ironjr/BlazingDriveV11m'
102
+ opt.model = 'KBlueLeaf/kohaku-v2.1'
103
  else:
104
  if opt.model.endswith('.safetensors'):
105
  opt.model = os.path.abspath(os.path.join('checkpoints', opt.model))