cheesyFishes commited on
Commit
07a2b5d
·
verified ·
1 Parent(s): bb8eae5

remove trust_remote_code

Browse files
Files changed (1) hide show
  1. custom_st.py +3 -0
custom_st.py CHANGED
@@ -31,6 +31,9 @@ class Transformer(nn.Module):
31
  self.dimension = dimension
32
  self.max_pixels = max_pixels
33
  self.min_pixels = min_pixels
 
 
 
34
 
35
  # Try to use flash attention if available, fallback to default attention if not
36
  try:
 
31
  self.dimension = dimension
32
  self.max_pixels = max_pixels
33
  self.min_pixels = min_pixels
34
+
35
+ # trust_remote_code is not needed here
36
+ kwargs.pop("trust_remote_code")
37
 
38
  # Try to use flash attention if available, fallback to default attention if not
39
  try: