fix added
Browse files- SkinGPT.py +1 -0
SkinGPT.py
CHANGED
|
@@ -16,6 +16,7 @@ import warnings
|
|
| 16 |
MODEL_DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 17 |
token = os.getenv("HF_TOKEN")
|
| 18 |
import streamlit as st
|
|
|
|
| 19 |
|
| 20 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 21 |
class Blip2QFormer(nn.Module):
|
|
|
|
| 16 |
MODEL_DTYPE = torch.float16 if torch.cuda.is_available() else torch.float32
|
| 17 |
token = os.getenv("HF_TOKEN")
|
| 18 |
import streamlit as st
|
| 19 |
+
import torch.nn.functional as F
|
| 20 |
|
| 21 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
| 22 |
class Blip2QFormer(nn.Module):
|