ejschwartz commited on
Commit
59324e3
·
1 Parent(s): f69c2e6

Logging fix

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -5,15 +5,12 @@ import frontmatter
5
  import os
6
  import spaces
7
  import torch
 
8
  from transformers import AutoTokenizer, AutoModelForCausalLM
9
- from accelerate import PartialState
10
- from accelerate.logging import get_logger
11
 
12
- # Initialize accelerate state first
13
- PartialState()
14
-
15
- # Enable accelerate logging
16
- logger = get_logger(__name__, log_level="DEBUG")
17
 
18
  import huggingface_hub
19
 
 
5
  import os
6
  import spaces
7
  import torch
8
+ import logging
9
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
10
 
11
+ # Set up standard logging
12
+ logging.basicConfig(level=logging.DEBUG)
13
+ logger = logging.getLogger(__name__)
 
 
14
 
15
  import huggingface_hub
16