Spaces:

DexterSptizu
/

SmolLM2-1.7B-Instruct

Runtime error

DexterSptizu commited on 6 days ago

Commit

ddd1c57

•

1 Parent(s): 9c49efc

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ class ModelHandler:
     def __init__(self):
         self.current_model = None
         self.current_tokenizer = None
-        self.device = "cuda" if torch.cuda.is_available() else "cpu"
     def load_model(self, model_name):
         try:

     def __init__(self):
         self.current_model = None
         self.current_tokenizer = None
+        self.device = "cpu" if torch.cuda.is_available() else "cpu"
     def load_model(self, model_name):
         try: