mirror of
https://github.com/nod-ai/AMD-SHARK-Studio.git
synced 2026-04-03 03:00:17 -04:00
Running SharkInference on CPUs, GPUs and MAC.
Run the binary sequence_classification.
The models supported are: hugging face sequence classification
./seq_classification.py --hf_model_name="hf_model" --device="cpu" # Use gpu | vulkan
Once the model is compiled to run on the device mentioned, we can pass in text and get the logits.