Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -48,6 +48,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
48 |
MODEL_NAME, trust_remote_code=True, device_map="auto"
|
49 |
)
|
50 |
model = model.eval()
|
|
|
51 |
|
52 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
53 |
print(f"Model device: {model.device}")
|
|
|
48 |
MODEL_NAME, trust_remote_code=True, device_map="auto"
|
49 |
)
|
50 |
model = model.eval()
|
51 |
+
print('model loadeddddddddddddddddddddddd')
|
52 |
|
53 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
54 |
print(f"Model device: {model.device}")
|