Update vision.py
This commit is contained in:
parent
5a5cfc70a1
commit
49b053c844
1 changed files with 0 additions and 2 deletions
|
|
@ -435,7 +435,6 @@ class FastBaseModel:
|
|||
if do_forced_float32: torch_dtype = torch.bfloat16
|
||||
|
||||
raise_handler = RaiseUninitialized()
|
||||
print(kwargs, model_name)
|
||||
model = auto_model.from_pretrained(
|
||||
model_name,
|
||||
device_map = device_map,
|
||||
|
|
@ -446,7 +445,6 @@ class FastBaseModel:
|
|||
# attn_implementation = attn_implementation,
|
||||
**kwargs,
|
||||
)
|
||||
print(model)
|
||||
raise_handler.remove()
|
||||
# Return old flag
|
||||
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = old_hf_transfer
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue