Spaces:

warhawkmonk
/

tiny-model

Running

warhawkmonk commited on 14 days ago

Commit

4392dee

verified ·

1 Parent(s): da68614

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 from transformers import pipeline
 # Model ID for Llama 3 8B instruct (replace with the exact model you want)
-MODEL_ID = "rumb/nano-mistral"
 # Load the text-generation pipeline with device_map="auto" to use GPU if available
 generator = pipeline(

 from transformers import pipeline
 # Model ID for Llama 3 8B instruct (replace with the exact model you want)
+MODEL_ID = "manycore-research/SpatialLM-Llama-1B"
 # Load the text-generation pipeline with device_map="auto" to use GPU if available
 generator = pipeline(