Paper99 commited on
Commit
fb796b9
β€’
1 Parent(s): e16bad8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ ASPECT_RATIO_LABELS = list(aspect_ratios)
40
  DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]
41
 
42
  enable_doodle_arg = False
43
- photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker", filename="photomaker-v2.bin", repo_type="model")
44
 
45
  if torch.cuda.is_available() and torch.cuda.is_bf16_supported():
46
  torch_dtype = torch.bfloat16
 
40
  DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]
41
 
42
  enable_doodle_arg = False
43
+ photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker-V2", filename="photomaker-v2.bin", repo_type="model")
44
 
45
  if torch.cuda.is_available() and torch.cuda.is_bf16_supported():
46
  torch_dtype = torch.bfloat16