Hugging Face's logo

HuggingFaceTB
/
SmolLM-135M-Instruct

Text Generation
Transformers
ONNX
Safetensors
English
llama
alignment-handbook
trl
sft
conversational
text-generation-inference
Inference Endpoints
Model card Files Community
14

Update code example in readme to use `device` variable instead of hard coded `cuda`

#4
by hl-tburns - opened
base: refs/heads/main
←
from: refs/pr/4
Discussion Files changed
+1
-1
hl-tburns

The code example uses device for model creation but then uses a hard coded cuda for inputs. Updated to use device for inputs as well.

Update code example in readme to use `device` variable instead of hard coded `cuda`8d040406
Xenova changed pull request status to merged
Your need to confirm your account before you can post a new comment.

Sign up or log in to comment