cheesyFishes
commited on
remove trust_remote_code
Browse files- 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:
|