More patches and Import IPEX after Torch

This commit is contained in:
Disty0
2023-04-30 18:19:37 +03:00
parent a720a670e8
commit de8d0bef9f
44 changed files with 202 additions and 24 deletions
+4
View File
@@ -3,6 +3,10 @@
import math
import numpy as np
import torch
try:
import intel_extension_for_pytorch as ipex
except:
pass
from torch import nn, Tensor
import torch.nn.functional as F
from typing import Optional, List
+4
View File
@@ -7,6 +7,10 @@ https://github.com/samb-t/unleashing-transformers/blob/master/models/vqgan.py
'''
import numpy as np
import torch
try:
import intel_extension_for_pytorch as ipex
except:
pass
import torch.nn as nn
import torch.nn.functional as F
import copy