sonoisa commited on
Commit
443e6f8
·
1 Parent(s): f5d1536

Remove unused model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -298,7 +298,7 @@ class ClipTextModel(nn.Module):
298
  # self.vision_model.save(os.path.join(output_dir, "vision_model"))
299
 
300
 
301
- def DummyClipModel:
302
  def __init__(self, text_model)
303
  self.text_model = text_model
304
 
 
298
  # self.vision_model.save(os.path.join(output_dir, "vision_model"))
299
 
300
 
301
+ class DummyClipModel:
302
  def __init__(self, text_model)
303
  self.text_model = text_model
304