Hugging Face's logo

Spaces:
venkat-natchi
/
AnyModeAssistant
Sleeping

App Files Community
venkat-natchi commited on
Commit
9b7046c
·
verified ·
1 Parent(s): e27df4d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -45,7 +45,6 @@ text_tokenizer = AutoTokenizer.from_pretrained(
45
  with torch.no_grad():
46
  tuned_phi2 = AutoModelForCausalLM.from_pretrained(
47
  "stage2_adaptor", trust_remote_code=True,
48
- device_map="auto",
49
  )
50
  base_phi2_text = AutoModelForCausalLM.from_pretrained(
51
  phi_model_name, trust_remote_code=True,
 
45
  with torch.no_grad():
46
  tuned_phi2 = AutoModelForCausalLM.from_pretrained(
47
  "stage2_adaptor", trust_remote_code=True,
 
48
  )
49
  base_phi2_text = AutoModelForCausalLM.from_pretrained(
50
  phi_model_name, trust_remote_code=True,