gdnartea commited on
Commit
dbabac6
·
verified ·
1 Parent(s): d6a9ee2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from nemo.collections.asr.models import ASRModel
13
  from nemo.collections.asr.parts.utils.streaming_utils import FrameBatchMultiTaskAED
14
  from nemo.collections.asr.parts.utils.transcribe_utils import get_buffered_pred_feat_multitaskAED
15
 
16
-
17
  model_id = "NousResearch/Meta-Llama-3-8B"
18
 
19
  llama_pipe = pipeline(
@@ -22,6 +22,16 @@ llama_pipe = pipeline(
22
  model_kwargs={"torch_dtype": torch.bfloat16},
23
  device_map="auto",
24
  )
 
 
 
 
 
 
 
 
 
 
25
 
26
 
27
  '''
 
13
  from nemo.collections.asr.parts.utils.streaming_utils import FrameBatchMultiTaskAED
14
  from nemo.collections.asr.parts.utils.transcribe_utils import get_buffered_pred_feat_multitaskAED
15
 
16
+ '''
17
  model_id = "NousResearch/Meta-Llama-3-8B"
18
 
19
  llama_pipe = pipeline(
 
22
  model_kwargs={"torch_dtype": torch.bfloat16},
23
  device_map="auto",
24
  )
25
+ '''
26
+
27
+
28
+ torch.random.manual_seed(0)
29
+ proc_model_name = "NousResearch/Meta-Llama-3-8B"
30
+ proc_model = AutoModelForCausalLM.from_pretrained(proc_model_name)
31
+ proc_model.to("cpu")
32
+ proc_tokenizer = AutoTokenizer.from_pretrained(proc_model_name)
33
+
34
+
35
 
36
 
37
  '''