Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
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 |
-
|
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))
|