Update vision.py

This commit is contained in:
Daniel Han 2025-03-16 23:20:43 -07:00
commit 5f841dd1e1

View file

@ -281,7 +281,6 @@ class FastBaseModel:
# Cannot be None, since HF now checks for the config
if load_in_4bit: kwargs["quantization_config"] = bnb_config
print(model_name)
model = auto_model.from_pretrained(
model_name,
device_map = device_map,