Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -9,9 +9,15 @@ MODEL_NAME = "mistralai/Mistral-7B-v0.1" # Hugging Face model adı
|
|
9 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
10 |
|
11 |
# === 2️⃣ CPU OPTİMİZASYONU ===
|
|
|
|
|
|
|
12 |
zero = torch.Tensor([0]).cuda()
|
13 |
print(zero.device) # <-- 'cpu' 🤔
|
14 |
-
|
|
|
|
|
|
|
15 |
|
16 |
print(zero.device) # <-- 'cuda:0' 🤗
|
17 |
|
|
|
9 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
10 |
|
11 |
# === 2️⃣ CPU OPTİMİZASYONU ===
|
12 |
+
|
13 |
+
|
14 |
+
|
15 |
zero = torch.Tensor([0]).cuda()
|
16 |
print(zero.device) # <-- 'cpu' 🤔
|
17 |
+
@spaces.GPU
|
18 |
+
def greet(n):
|
19 |
+
print(zero.device) # <-- 'cuda:0' 🤗
|
20 |
+
return f"Hello {zero + n} Tensor"
|
21 |
|
22 |
print(zero.device) # <-- 'cuda:0' 🤗
|
23 |
|