AlekseyCalvin commited on
Commit
197711b
·
verified ·
1 Parent(s): a272613

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,11 +52,11 @@ torch_dtype=torch.bfloat16,
52
  trust_remote_code=True,
53
  ).to(device)
54
 
55
- clipmodel = 'long'
56
  if clipmodel == "long":
57
  model_id = "zer0int/LongCLIP-GmP-ViT-L-14"
58
  config = CLIPConfig.from_pretrained(model_id)
59
- maxtokens = 248
60
  if clipmodel == "norm":
61
  model_id = "zer0int/CLIP-GmP-ViT-L-14"
62
  config = CLIPConfig.from_pretrained(model_id)
 
52
  trust_remote_code=True,
53
  ).to(device)
54
 
55
+ clipmodel = 'norm'
56
  if clipmodel == "long":
57
  model_id = "zer0int/LongCLIP-GmP-ViT-L-14"
58
  config = CLIPConfig.from_pretrained(model_id)
59
+ maxtokens = 77
60
  if clipmodel == "norm":
61
  model_id = "zer0int/CLIP-GmP-ViT-L-14"
62
  config = CLIPConfig.from_pretrained(model_id)