wangpangintsig commited on
Commit
ef5f6cd
·
verified ·
1 Parent(s): 5dd409e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=True, **hf_kwargs)
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