Spaces:
Runtime error
Runtime error
Commit
·
350967e
1
Parent(s):
a8635dd
Update open_flamingo/open_flamingo/src/factory.py
Browse files
open_flamingo/open_flamingo/src/factory.py
CHANGED
|
@@ -1,5 +1,6 @@
|
|
| 1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import open_clip
|
|
|
|
| 3 |
|
| 4 |
from .flamingo import Flamingo
|
| 5 |
from .flamingo_lm import FlamingoLMMixin
|
|
|
|
| 1 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 2 |
import open_clip
|
| 3 |
+
import torch
|
| 4 |
|
| 5 |
from .flamingo import Flamingo
|
| 6 |
from .flamingo_lm import FlamingoLMMixin
|