ResearcherXman commited on
Commit
6aae2cc
1 Parent(s): 857a5c4
Files changed (1) hide show
  1. app.py +1 -0
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