prithivMLmods commited on
Commit
ed8e40d
·
verified ·
1 Parent(s): 6d1a1b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
31
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
32
 
33
  # Load text-only model and tokenizer
34
- model_id = "prithivMLmods/FastThink-0.5B-Tiny"
35
  tokenizer = AutoTokenizer.from_pretrained(model_id)
36
  model = AutoModelForCausalLM.from_pretrained(
37
  model_id,
 
31
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
32
 
33
  # Load text-only model and tokenizer
34
+ model_id = "prithivMLmods/DeepHermes-3-Llama-3-3B-Preview-abliterated"
35
  tokenizer = AutoTokenizer.from_pretrained(model_id)
36
  model = AutoModelForCausalLM.from_pretrained(
37
  model_id,