Hugging Face
Models
Datasets
Spaces
Posts
Docs
Enterprise
Pricing
Log In
Sign Up
Spaces:
orby-osu
/
UGround
like
13
Sleeping
App
Files
Files
Community
1
20b85f0
UGround
/
llava
/
train
/
train_mem.py
BoyuNLP
init
3bbba47
3 months ago
raw
Copy download link
history
blame
Safe
115 Bytes
from
llava.train.train
import
train
if
__name__ ==
"__main__"
:
train(attn_implementation=
"flash_attention_2"
)