Spaces:
Sleeping
Sleeping
ADD gpu memory
Browse files
app.py
CHANGED
|
@@ -90,7 +90,7 @@ prompt_model,_,_ = open_clip.create_model_and_transforms('ViT-bigG-14', 'laion2b
|
|
| 90 |
prompt_model = prompt_model.to(device)
|
| 91 |
prompt_model.eval()
|
| 92 |
tokenizer = open_clip.get_tokenizer('ViT-bigG-14')
|
| 93 |
-
|
| 94 |
|
| 95 |
|
| 96 |
|
|
|
|
| 90 |
prompt_model = prompt_model.to(device)
|
| 91 |
prompt_model.eval()
|
| 92 |
tokenizer = open_clip.get_tokenizer('ViT-bigG-14')
|
| 93 |
+
print("GPU memory:",torch.cuda.memory_allocated(0))
|
| 94 |
|
| 95 |
|
| 96 |
|