Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -15,7 +15,7 @@ class HFEmbedder(nn.Module):
|
|
| 15 |
self.hf_module: CLIPTextModel = CLIPTextModel.from_pretrained(version, **hf_kwargs)
|
| 16 |
else:
|
| 17 |
self.tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(version, max_length=max_length)
|
| 18 |
-
self.hf_module: T5EncoderModel = T5EncoderModel.from_pretrained(version, use_safetensors=
|
| 19 |
|
| 20 |
self.hf_module = self.hf_module.eval().requires_grad_(False)
|
| 21 |
|
|
|
|
| 15 |
self.hf_module: CLIPTextModel = CLIPTextModel.from_pretrained(version, **hf_kwargs)
|
| 16 |
else:
|
| 17 |
self.tokenizer: T5Tokenizer = T5Tokenizer.from_pretrained(version, max_length=max_length)
|
| 18 |
+
self.hf_module: T5EncoderModel = T5EncoderModel.from_pretrained(version, use_safetensors=False, **hf_kwargs)
|
| 19 |
|
| 20 |
self.hf_module = self.hf_module.eval().requires_grad_(False)
|
| 21 |
|