oriolgds commited on
Commit
c082eba
·
unverified ·
1 Parent(s): b7e1b4b

Revert "Changed to 1B"

Browse files

This reverts commit b7e1b4b8d6430aa710878a7e7fec12ffcc127942.

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from datetime import datetime
13
 
14
  class LlamaChat:
15
  def __init__(self):
16
- self.model_name = "meta-llama/Llama-3.2-1B-Instruct"
17
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
18
  self.tokenizer = None
19
  self.model = None
 
13
 
14
  class LlamaChat:
15
  def __init__(self):
16
+ self.model_name = "meta-llama/Llama-3.2-3B-Instruct"
17
  self.device = "cuda" if torch.cuda.is_available() else "cpu"
18
  self.tokenizer = None
19
  self.model = None