Spaces:

marcelbinz
/

Centaur

Runtime error

marcelbinz commited on Jun 30

Commit

85e8a39

verified ·

1 Parent(s): 7173b8f

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import gradio as gr
 import torch
 from transformers import pipeline, BitsAndBytesConfig, AutoModelForCausalLM, AutoTokenizer
-MODEL_ID = "marcelbinz/Llama-3.1-Centaur-70B"
 bnb_4bit_config = BitsAndBytesConfig(
     load_in_4bit=True,

 import torch
 from transformers import pipeline, BitsAndBytesConfig, AutoModelForCausalLM, AutoTokenizer
+MODEL_ID = "marcelbinz/Llama-3.1-Minitaur-8B"
 bnb_4bit_config = BitsAndBytesConfig(
     load_in_4bit=True,