Hugging Face's logo

Spaces:
manindragautam
/
gemma-text-encoder
Runtime error

App Files Community
linoyts HF Staff commited on
Commit
c6ba0d3
·
verified ·
1 Parent(s): 8df62d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ print("Loading Gemma Text Encoder...")
34
  print("=" * 80)
35
 
36
  checkpoint_path = get_hub_or_local_checkpoint(DEFAULT_REPO_ID, DEFAULT_CHECKPOINT_FILENAME)
37
- device = get_device()
38
 
39
  print(f"Initializing text encoder with:")
40
  print(f" checkpoint_path={checkpoint_path}")
 
34
  print("=" * 80)
35
 
36
  checkpoint_path = get_hub_or_local_checkpoint(DEFAULT_REPO_ID, DEFAULT_CHECKPOINT_FILENAME)
37
+ device = "cuda"
38
 
39
  print(f"Initializing text encoder with:")
40
  print(f" checkpoint_path={checkpoint_path}")