Update models/epalm.py

models/epalm.py

@@ -211,7 +211,7 @@ class ePALM(nn.Module):
         self.no_attention_mask = False
 
         if low_cpu:
-            self.model_text = OPTForCausalLM.from_pretrained(opt_model_name, config=config_opt, revision="float16", torch_dtype=torch.float16, low_cpu_mem_usage=False)
+            self.model_text = OPTForCausalLM.from_pretrained(opt_model_name, config=config_opt, torch_dtype=torch.float16, low_cpu_mem_usage=False)
         else:
             self.model_text = OPTForCausalLM.from_pretrained(opt_model_name, config=config_opt)