aladdin1995 commited on
Commit
d84e57e
·
verified ·
1 Parent(s): d400d0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ def replace_single_quotes(text):
27
  return replaced_text
28
 
29
  class PromptEnhancerV2:
30
- def __init__(self, models_root_path, device_map="auto", torch_dtype="bfloat16"):
31
  if not logging.getLogger(__name__).handlers:
32
  logging.basicConfig(level=logging.INFO)
33
  self.logger = logging.getLogger(__name__)
 
27
  return replaced_text
28
 
29
  class PromptEnhancerV2:
30
+ def __init__(self, models_root_path, device_map="cuda", torch_dtype="bfloat16"):#auto
31
  if not logging.getLogger(__name__).handlers:
32
  logging.basicConfig(level=logging.INFO)
33
  self.logger = logging.getLogger(__name__)