Update vision.py
This commit is contained in:
parent
0639a31a6c
commit
53153eaab5
1 changed files with 0 additions and 1 deletions
|
|
@ -408,7 +408,6 @@ class FastBaseModel:
|
|||
torch_dtype = dtype
|
||||
if do_forced_float32: torch_dtype = torch.bfloat16
|
||||
|
||||
print(kwargs)
|
||||
model = auto_model.from_pretrained(
|
||||
model_name,
|
||||
device_map = device_map,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue