Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ class _MLPVectorProjector(nn.Module):
|
|
28 |
model_name = "microsoft/phi-2"
|
29 |
|
30 |
with torch.no_grad():
|
31 |
-
phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="auto"
|
32 |
|
33 |
tokenizer_text = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
34 |
|
|
|
28 |
model_name = "microsoft/phi-2"
|
29 |
|
30 |
with torch.no_grad():
|
31 |
+
phi2_text = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True, device_map="auto")
|
32 |
|
33 |
tokenizer_text = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
|
34 |
|