Spaces:

khazarai
/

Bio-Reasoning

Sleeping

Rustamshry commited on 16 days ago

Commit

4830f4a

verified ·

1 Parent(s): 0ae68e6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ tokenizer = AutoTokenizer.from_pretrained("khazarai/BioGenesis-ToT")
 model = AutoModelForCausalLM.from_pretrained(
     "khazarai/BioGenesis-ToT",
     dtype=torch.float32,
-    device_map={"": "cpu"},
 )
 # --- Chatbot logic ---

 model = AutoModelForCausalLM.from_pretrained(
     "khazarai/BioGenesis-ToT",
     dtype=torch.float32,
+    #device_map={"": "cpu"},
 )
 # --- Chatbot logic ---