Spaces:

ejschwartz
/

test

Runtime error

ejschwartz commited on Jun 9

Commit

f69c2e6

1 Parent(s): de7cd64

Logging fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,12 @@ import os
 import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 from accelerate.logging import get_logger
 # Enable accelerate logging
 logger = get_logger(__name__, log_level="DEBUG")

 import spaces
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from accelerate import PartialState
 from accelerate.logging import get_logger
+# Initialize accelerate state first
+PartialState()
 # Enable accelerate logging
 logger = get_logger(__name__, log_level="DEBUG")