[Fix] Fix quoted strings

pull/1700/head
fanqiNO1 2023-07-10 11:53:06 +08:00
parent 5bc53aea8e
commit 28a769f78a
1 changed files with 3 additions and 3 deletions

View File

@ -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)