Spaces:
Running
on
A10G
Running
on
A10G
Commit
•
68143b2
1
Parent(s):
fecea78
Update minigpt4/models/mini_gpt4.py
Browse files
minigpt4/models/mini_gpt4.py
CHANGED
|
@@ -6,7 +6,7 @@
|
|
| 6 |
"""
|
| 7 |
import logging
|
| 8 |
import random
|
| 9 |
-
|
| 10 |
import torch
|
| 11 |
from torch.cuda.amp import autocast as autocast
|
| 12 |
import torch.nn as nn
|
|
|
|
| 6 |
"""
|
| 7 |
import logging
|
| 8 |
import random
|
| 9 |
+
import os
|
| 10 |
import torch
|
| 11 |
from torch.cuda.amp import autocast as autocast
|
| 12 |
import torch.nn as nn
|