Remove unused model
Browse files
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 |
-
|
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 |
|