[Fix] Fix quoted strings
parent
5bc53aea8e
commit
28a769f78a
|
@ -165,7 +165,7 @@ class Blip2Caption(BaseModel):
|
|||
attns_opt = torch.ones(
|
||||
inputs_opt.size()[:-1], dtype=torch.long).to(images.device)
|
||||
|
||||
self.tokenizer.padding_side = "right"
|
||||
self.tokenizer.padding_side = 'right'
|
||||
|
||||
prompt = [
|
||||
data_sample.gt_caption + '\n' for data_sample in data_samples
|
||||
|
@ -174,7 +174,7 @@ class Blip2Caption(BaseModel):
|
|||
opt_tokens = self.tokenizer(
|
||||
prompt,
|
||||
return_tensors='pt',
|
||||
padding="longest",
|
||||
padding='longest',
|
||||
truncation=True,
|
||||
max_length=self.max_txt_len,
|
||||
).to(images.device)
|
||||
|
@ -248,7 +248,7 @@ class Blip2Caption(BaseModel):
|
|||
opt_tokens = self.tokenizer(
|
||||
prompt,
|
||||
return_tensors='pt',
|
||||
padding="longest",
|
||||
padding='longest',
|
||||
truncation=True,
|
||||
max_length=self.max_txt_len,
|
||||
).to(images.device)
|
||||
|
|
Loading…
Reference in New Issue