Skip to content

Commit f876bdb

Browse files
committed
No longer require trust_remote_code
1 parent 7cdec11 commit f876bdb

2 files changed

Lines changed: 1 addition & 3 deletions

File tree

example_florence2_usage.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@ def main():
1717
llm = LLM(
1818
model=model_name,
1919
mm_processor_cache_gb=0,
20-
trust_remote_code=True,
2120
enforce_eager=True,
2221
)
2322
params = SamplingParams(

tests/test_florence2.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,6 @@ def florence2_llm():
131131
os.environ["VLLM_ENABLE_V1_MULTIPROCESSING"] = "0"
132132
return LLM(
133133
model=MODEL_NAME,
134-
trust_remote_code=True,
135134
enforce_eager=True,
136135
gpu_memory_utilization=0.5,
137136
mm_processor_cache_gb=0,
@@ -203,7 +202,7 @@ def test_encoder_length_within_limit(self, stop_sign_image):
203202
"""Processor output must not exceed BART max_position_embeddings."""
204203
from transformers import AutoProcessor
205204

206-
processor = AutoProcessor.from_pretrained(MODEL_NAME, trust_remote_code=True)
205+
processor = AutoProcessor.from_pretrained(MODEL_NAME)
207206
out = processor(
208207
text="<DETAILED_CAPTION>", images=stop_sign_image, return_tensors="pt"
209208
)

0 commit comments

Comments
 (0)