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
|