Spaces:

helloperson123
/

tiny-llama-chatbot

Runtime error

helloperson123 commited on Jan 10

Commit

598a5d4

verified ·

1 Parent(s): 1ffd59d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,6 @@ DEVICE = "cpu"
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.float32
-)
 model.to(DEVICE)
 )

 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.float32
 model.to(DEVICE)
 )