Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -20,7 +20,7 @@ moondream = AutoModelForCausalLM.from_pretrained(
|
|
| 20 |
)
|
| 21 |
moondream.eval()
|
| 22 |
|
| 23 |
-
control_vectors = torch.load("control_vectors.pt")
|
| 24 |
control_vectors = [t.to('cuda', dtype=torch.bfloat16) for t in control_vectors]
|
| 25 |
|
| 26 |
class LayerWrapper(torch.nn.Module):
|
|
|
|
| 20 |
)
|
| 21 |
moondream.eval()
|
| 22 |
|
| 23 |
+
control_vectors = torch.load("control_vectors.pt", map_location="cpu")
|
| 24 |
control_vectors = [t.to('cuda', dtype=torch.bfloat16) for t in control_vectors]
|
| 25 |
|
| 26 |
class LayerWrapper(torch.nn.Module):
|