Tomtom84 commited on
Commit
6214f63
·
verified ·
1 Parent(s): 1b79dec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -22,8 +22,9 @@ device = "cpu"
22
  print("Loading SNAC model...")
23
  snac = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").to(device)
24
 
25
- model_name = "canopylabs/3b-de-ft-research_release"
26
  # optional: explizites snapshot_download (entfernt große Dateien)
 
27
  snapshot_download(
28
  repo_id=model_name,
29
  allow_patterns=["config.json", "*.safetensors", "model.safetensors.index.json"],
@@ -33,7 +34,7 @@ snapshot_download(
33
  "special_tokens_map.json", "vocab.json", "merges.txt", "tokenizer.*"
34
  ]
35
  )
36
-
37
  print("Loading Orpheus model...")
38
  model = AutoModelForCausalLM.from_pretrained(
39
  model_name, torch_dtype=torch.bfloat16
 
22
  print("Loading SNAC model...")
23
  snac = SNAC.from_pretrained("hubertsiuzdak/snac_24khz").to(device)
24
 
25
+ model_name = "canopylabs/3b-de-pretrain-research_release"
26
  # optional: explizites snapshot_download (entfernt große Dateien)
27
+ '''
28
  snapshot_download(
29
  repo_id=model_name,
30
  allow_patterns=["config.json", "*.safetensors", "model.safetensors.index.json"],
 
34
  "special_tokens_map.json", "vocab.json", "merges.txt", "tokenizer.*"
35
  ]
36
  )
37
+ '''
38
  print("Loading Orpheus model...")
39
  model = AutoModelForCausalLM.from_pretrained(
40
  model_name, torch_dtype=torch.bfloat16