TinyLlama icon indicating copy to clipboard operation
TinyLlama copied to clipboard

Is it supported to convert to Apple CoreML? I tried to use coremltools to convert this model to CoreML format, but encountered an error.

Open HoodL opened this issue 2 years ago • 1 comments
trafficstars

The error message is as follows: raise NotSupportedError(r, "Comprehension ifs are not supported yet") torch.jit.frontend.NotSupportedError: Comprehension ifs are not supported yet: File "/Users/Lucas/Library/Python/3.9/lib/python/site-packages/transformers/models/llama/modeling_llama.py", line 947 next_cache = next_decoder_cache if use_cache else None if not return_dict: return tuple(v for v in [hidden_states, next_cache, all_hidden_states, all_self_attns] if v is not None) return BaseModelOutputWithPast( last_hidden_state=hidden_states,

HoodL avatar Nov 21 '23 06:11 HoodL

This project was meant to be trained and inferenced on a GPU. It does extend to CPU via GGUF, but there is no real support for CoreML/Metal, beyond the standard levels from lit-gpt

VatsaDev avatar Nov 22 '23 22:11 VatsaDev