hidehisa-arai commited on
Commit
1e586a4
1 Parent(s): e7036b0

debug model

Browse files
Files changed (1) hide show
  1. modeling_japanese_clip.py +1 -1
modeling_japanese_clip.py CHANGED
@@ -206,7 +206,7 @@ class Transformer(nn.Module):
206
  return x
207
 
208
 
209
- class JapaneseCLIPVisionTransformer(PreTrainedModel):
210
  output_tokens: torch.jit.Final[bool]
211
 
212
  def __init__(
 
206
  return x
207
 
208
 
209
+ class JapaneseCLIPVisionTransformer(nn.Module):
210
  output_tokens: torch.jit.Final[bool]
211
 
212
  def __init__(