ResearcherXman
commited on
Commit
•
6aae2cc
1
Parent(s):
857a5c4
update
Browse files
app.py
CHANGED
@@ -55,6 +55,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
55 |
dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
|
56 |
STYLE_NAMES = list(styles.keys())
|
57 |
DEFAULT_STYLE_NAME = "Watercolor"
|
|
|
58 |
|
59 |
# download checkpoints
|
60 |
from huggingface_hub import hf_hub_download
|
|
|
55 |
dtype = torch.float16 if str(device).__contains__("cuda") else torch.float32
|
56 |
STYLE_NAMES = list(styles.keys())
|
57 |
DEFAULT_STYLE_NAME = "Watercolor"
|
58 |
+
enable_lcm_arg = False
|
59 |
|
60 |
# download checkpoints
|
61 |
from huggingface_hub import hf_hub_download
|