yujiepan commited on
Commit
fcef870
·
verified ·
1 Parent(s): 0c24726

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,7 +19,7 @@ import torch
19
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
20
  torch.random.manual_seed(0)
21
 
22
- model_id = "tiny-random/phi4-flash"
23
  model = AutoModelForCausalLM.from_pretrained(
24
  model_id,
25
  device_map="cuda",
@@ -69,7 +69,7 @@ from transformers import (
69
  )
70
 
71
  source_model_id = "microsoft/Phi-4-mini-flash-reasoning"
72
- save_folder = "/tmp/tiny-random/phi4-flash"
73
 
74
  processor = AutoProcessor.from_pretrained(source_model_id, trust_remote_code=True)
75
  processor.save_pretrained(save_folder)
 
19
  from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
20
  torch.random.manual_seed(0)
21
 
22
+ model_id = "tiny-random/phi-4-flash"
23
  model = AutoModelForCausalLM.from_pretrained(
24
  model_id,
25
  device_map="cuda",
 
69
  )
70
 
71
  source_model_id = "microsoft/Phi-4-mini-flash-reasoning"
72
+ save_folder = "/tmp/tiny-random/phi-4-flash"
73
 
74
  processor = AutoProcessor.from_pretrained(source_model_id, trust_remote_code=True)
75
  processor.save_pretrained(save_folder)